search

OpenGVLab / VideoMamba

VideoMamba: State Space Model for Efficient Video Understanding
https://arxiv.org/abs/2403.06977
Apache License 2.0
660 stars 47 forks source link

I found an error for videoamba_pretrain.py #27

Closed jlu-phyComputer closed 1 month ago

jlu-phyComputer commented 2 months ago

I'm trying to run videoamba_pretrain.py But I received the following error: Traceback (most recent call last): File "/home/zhangjianqi/workspace/python_code/VideoMamba main/videoamba/video_sm/models/videoamba_pretrain. py", line 474, in Model=videoamba_middle_pretrain (numframes=numframes). cuda() File "/home/zhangjanqi/workspace/python_code/VideoMamba main/videoamba/video_sm/models/videoamba_pretrain. py", line 450, in videoamba_middle_pretrain Model=VisionMamba（ File "/home/zhangjianqi/workspace/python_code/VideoMamba main/videoamba/video_sm/models/videoamba_pretrain. py", line 304, in init [ File "/home/zhangjanqi/workspace/python_code/VideoMamba main/videoamba/video_sm/models/videoamba_pretrain. py", line 305, in Createblock（ File "/home/zhangjianqi/workspace/python_code/VideoMamba main/videoamba/video_sm/models/videoamba_pretrain. py", line 112, in createblock Block=Block（ File "/home/zhangjanqi/workspace/python_code/VideoMamba main/videoamba/video_sm/models/videoamba_pretrain. py", line 48, in init Self. mixer=mixer_cls (dim) TypeError: Mamba Init () got an unexpected keyword argument 'bimamba' Use checkpoint: False Checkpoint number: 0 Student return index: [31] Process completed with exit code 1

Andy1621 commented 2 months ago

Hi! Please install the mamba package via pip install -e mamba.

jlu-phyComputer commented 2 months ago

Your requirement does not specify the version of Mamba. I installed mamba-ssh=1.2.0. post1

Andy1621 commented 2 months ago

Hi! The mamba can be found in our repo. You should complie it by yourself.

wjj-w commented 2 months ago

Your requirement does not specify the version of Mamba. I installed mamba-ssh=1.2.0. post1 Hello, I have the same problem. Have you solved the problem now?