search

MzeroMiko / VMamba

VMamba: Visual State Space Models，code is based on mamba
MIT License
1.82k stars 98 forks source link

Backbones #215

Open KosaAlex opened 4 weeks ago

KosaAlex commented 4 weeks ago

Could you please provide the backbones used for segmentation during training? I would want the one used for training VSSM small.

MzeroMiko commented 3 weeks ago

All the checkpoint can be found in releases, while the code lies in model.py in the folder segmentation.