weixin_47245869的博客

原创 解决windows版Vimamba运行main.py跑自己的数据集报错问题

分析错误原因，可能是作者在定义VisionMamba块的时候，用的是Visiontransformer框架，而。方法并不接受这两个参数，因此在调用模型时，将这两个参数给删掉了（有大佬说一下这样会有什么坏处吗？子目录下放图片，我是做图像分类，必须要按照train和val的数据分开放，不然会报错。等几篇博客，当然，在安装过程中也遇到了其他问题，但一步步终归是解决了。在此之前，最好先对数据预处理一下，将图片格式变为224*224大小。outputs = model(...)中。这两个参数是为vim定制的，