search

MzeroMiko / VMamba

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

About Visualization #256

Open xiequan277 opened 1 month ago

xiequan277 commented 1 month ago

Thanks for sharing your work, i can`t use attnmap.py The problem is :AttributeError: 'SS2D' object has no attribute 'data'. Did you mean: 'call'? please help me QAQ

MzeroMiko commented 1 month ago

some SS2Ds do not have data attribute, you may need to add the code yourself as shown in SS2Dv2:

https://github.com/MzeroMiko/VMamba/blob/0a74a29eefb9223efc1a399000e22a723390defd/classification/models/vmamba.py#L631