Vision Mamba复现过程及相关问题记录

记录在复现过程中自己遇到的问题及解决办法，仅供参考，目的只为跑起来，不纠结原因

代码地址：https://github.com/hustvl/Vim

论文地址：[2401.09417] Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model (arxiv.org)

环境要求

readme里的要求如下

• Python 3.10.13

  • conda create -n your_env_name python=3.10.13

• torch 2.1.1 + cu118

  • pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118

• Requirements: vim_requirements.txt

  • pip install -r vim/vim_requirements.txt

• Install causal_conv1d and mamba

  • pip install -e causal-conv1d>=1.1.0
  • pip install -e mamba-1p1p1

遇到的问题：

• 安装torch时查找不到版本，直接去PyTorch官网找到对应版本，使用conda安装成功

conda install pytorch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 pytorch-cuda=11.8 -c pytorch -c nvidia

• 如果直接用pip install时出现错误，可以查看python和pip版本是否一致 

(py10cu118) ds@ds-System-Product-Name:~$ python --version
Python 3.10.13
(py10cu118) ds@ds-System-Product-Name:~$ pip --version
pip 21.2.4 from /home/ds/anaconda3/envs/zlb/lib/python3.6/site-packages/pip (python 3.6)

可以在使用pip时在前面添加python -m

python -m pip install -r vim/vim_requirements.txt

关于mamba和casual-conv1d的安装也可直接见下面的参考文章

训练的时候需要数据集地址，我用的CIFAR，直接在dataset.py里把CIFAR里添加download=True

训练遇到如下问题：

TypeError: Mamba.__init__() got an unexpected keyword argument 'bimamba_type'

参考文章

复现Vision Mamba时报错“TypeError: causal_conv1d_fwd(): incompatible function arguments.”_typeerror: fwd(): incompatible function arguments.-CSDN博客