YuHengsss/MSVMamba

visualization

Closed this issue · 2 comments

Zizzzzzzz commented

Thank you for your excellent work. I would like to inquire about how to obtain the visualized results of Figures 2 and 4 from the paper.

YuHengsss commented

Thanks for your attention on our work. The code related to the visualization can be found here:
https://github.com/YuHengsss/MSVMamba/blob/master/analyze/attnmap.py

Zizzzzzzz commented

Thanks! I will try it.