How to add a mamba backbone as an encoder for training

[ljb-1]

No description provided.

[qubvel]

Hey @ljb-1! Thanks for the question, at the moment it's not supported out of the box. You can make your own backbone based on timm_universal and/or mix_transformer backbones. I'm going to add support for more timm models in coming months.

[qubvel]

Hey @ljb-1,

Mamba out encoders are supported with the latest main branch, check this out with the following encoders (add tu- prefix to names)

mambaout_base
mambaout_base_plus_rw
mambaout_base_short_rw
mambaout_base_tall_rw
mambaout_base_wide_rw
mambaout_femto
mambaout_kobe
mambaout_small
mambaout_small_rw
mambaout_tiny