MzeroMiko / VMamba

VMamba: Visual State Space Models,code is based on mamba
MIT License
1.83k stars 100 forks source link

If I want to add some module in vmamba.py, how to train it to get a new model? #179

Open lilitao0517 opened 2 months ago

lilitao0517 commented 2 months ago

after coding,i can train it and get .pth files,but when i test this model,he told me that:unexpected key in source state_dict: backbone.rgb_to_hsvs_attention.learnable_matrix,and rgb_to_hsvs_attention.learnable_matrix is my own module. Does this prove that the module I added myself was not trained (or even the model structure was updated). How to solve this problem?thank you!