复现Vision Mamba时报错“TypeError: causal_conv1d_fwd(): incompatible function arguments.”

一开始使用https://github.com/hustvl/Vim给出的方法配置环境

pip install -e causal_conv1d>=1.1.0
pip install -e mamba-1p1p1

训练模型时出现报错：

原因是causal_conv1d版本不对（使用pip install -e causal_conv1d>=1.1.0安装完版本为1.0.0，后安装mamba-ssm时被替换为1.2.0版本），需要重新安装。

方法：

删除环境

pip uninstall causal-conv1d
pip uninstall mamba-ssm

安装正确版本

pip install causal-conv1d==1.1.0
pip install mamba==1.1.1

注：安装causal-conv1d时出现网络问题需要手动安装

此时环境基本已经装好了，但是运行代码会遇到下面的这个问题

“TypeError: Mamba.init() got an unexpected keyword argument ‘bimamba_type’.”

此时需要将site-package下的mamba_ssm文件夹删除，并替换为github项目代码里的mamba_ssm文件夹Vim/mamba-1p1p1/mamba_ssm at main · hustvl/Vim (github.com)

rm -rf ".../python3.8/site-packages/mamba_ssm/" (删除环境下的mamba_ssm文件夹)
cp -r "Vim-main/mamba-1p1p1/mamba_ssm" ".../python3.8/site-packages/" (替换mamba_ssm文件夹)

运行成功。