执行预训练文件run_videomamba_pretraining.py一直报错,提示got an unexpected keyword argument 'checkpoint_num'
Opened this issue · 0 comments
Traceback (most recent call last): File "/home/mi/PycharmProjects/VideoMamba-main/videomamba/video_sm/run_videomamba_pretraining.py", line 353, in <module> main(opts) File "/home/mi/PycharmProjects/VideoMamba-main/videomamba/video_sm/run_videomamba_pretraining.py", line 202, in main model = get_model(args) File "/home/mi/PycharmProjects/VideoMamba-main/videomamba/video_sm/run_videomamba_pretraining.py", line 171, in get_model model = create_model( File "/home/mi/anaconda3/envs/videomamba3/lib/python3.10/site-packages/timm/models/factory.py", line 81, in create_model model = create_fn(pretrained=pretrained, **kwargs) File "/home/mi/PycharmProjects/VideoMamba-main/videomamba/video_sm/models/modeling_pretrain.py", line 282, in pretrain_videomae_base_patch16_224 model = PretrainVisionTransformer( TypeError: PretrainVisionTransformer.__init__() got an unexpected keyword argument 'checkpoint_num'
请问这个问题是出在代码问题上还是包的版本问题上