THU-MIG / RepViT

RepViT: Revisiting Mobile CNN From ViT Perspective [CVPR 2024] and RepViT-SAM: Towards Real-Time Segmenting Anything
https://arxiv.org/abs/2307.09283
Apache License 2.0
681 stars 55 forks source link

typeError: repvit_m0_9() got an unexpected keyword argument 'pretrained_cfg' #56

Closed wdqffywlw closed 3 months ago

wdqffywlw commented 3 months ago

我用的环境是wsl+ubuntu+vsconde,pytoch安装好了,repvit对应项目所需要的包也安装完毕。在vscode的terminal下,执行命令:

python --master_port 12346 main.py --model repvit_m0_9 --data-path ~/imagenet --dist-eval ,(因为只有一个gpu,去掉了参数 -m torch.distributed.launch --nproc_per_node=8 --master_port 12346 --use_env),运行后,出现提示: Creating model: repvit_m0_9 Traceback (most recent call last): File "/home/zhifan/project/RepViT/main.py", line 485, in main(args) File "/home/zhifan/project/RepViT/main.py", line 267, in main model = create_model( File "/home/zhifan/anaconda3/envs/pytorch/lib/python3.9/site-packages/timm/models/factory.py", line 71, in create_model model = create_fn(pretrained=pretrained, pretrained_cfg=pretrained_cfg, **kwargs) TypeError: repvit_m0_9() got an unexpected keyword argument 'pretrained_cfg' 这个问题该如何解决?