hunto / LocalMamba

Code for paper LocalMamba: Visual State Space Model with Windowed Selective Scan
Apache License 2.0
193 stars 10 forks source link

关于ckpt精度的疑问 #16

Open JRF0504 opened 5 months ago

JRF0504 commented 5 months ago

作者你好,我下载了你公布的两个ckpt文件,并在ImageNet上进行了测试。其中LocalVMamba-S得到了一致的83.7的准确率,但是LocalVim-T我测试得到的精度只有73.66,与你所描述的76.2有较大的差距,请问这是为什么呢?

hunto commented 5 months ago

哈喽,我刚刚用最新的代码和上传的ckpt测试了精度,没有出现你遇到的问题。可以提供更详细的信息帮助定位bug吗? localvim_tiny_val_log_20240419_204257.txt