search

MzeroMiko / VMamba

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

Configurations not Reflecting in Model Construction and Initialization in mmseg #201

Open ZhangIceNight opened 6 months ago

ZhangIceNight commented 6 months ago

Hello,

I'm glad you shared your work!

When I was reproducing the segmentation, I found that no matter how I modified the contents in the config file, the model actually built did not adopt the operations in the config file, but remained default.

For example, in the config you provided, forward_type="v05_noz", but I found through outputting this variable that forward_type="v2".

Furthermore, I gradually output and found that the __init__ method of the class Backbone_VSSM in the file vmamba.py seems not to be called.

I'm not very familiar with mmseg, and I hope to get your reply and help.

Thank you so much

MzeroMiko commented 6 months ago

The class Backbone_VSSM is called in segmentation/model.py by calling class MM_VSSM, which is registered into mmdet.registry and mmseg.registry.