RepViT: Revisiting Mobile CNN From ViT Perspective [CVPR 2024] and RepViT-SAM: Towards Real-Time Segmenting Anything
681
stars
55
forks
source link
typeError: repvit_m0_9() got an unexpected keyword argument 'pretrained_cfg' #56
Closed
wdqffywlw closed 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'
这个问题该如何解决?