alibaba/EasyCV

TypeError: create_model() got multiple values for argument 'pretrained'

zzoneee opened this issue · 0 comments

I encountered this problem while running the first example.

python tools/train.py configs/classification/cifar10/swintiny_b64_5e_jpg.py --work_dir work_dirs/classification/cifar10/swintiny --fp16

Traceback (most recent call last):
File "tools/train.py", line 277, in
main()
File "tools/train.py", line 213, in main
model = build_model(cfg.model)
File "/home/zhenzhen/ev/tmp/EasyCV/package/easycv/models/builder.py", line 39, in build_model
return build(cfg, MODELS)
File "/home/zhenzhen/ev/tmp/EasyCV/package/easycv/models/builder.py", line 15, in build
return build_from_cfg(cfg, registry, default_args)
File "/home/zhenzhen/ev/tmp/EasyCV/package/easycv/utils/registry.py", line 80, in build_from_cfg
return obj_cls(**args)
File "/home/zhenzhen/ev/tmp/EasyCV/package/easycv/models/classification/classification.py", line 77, in init
self.backbone = builder.build_backbone(backbone)
File "/home/zhenzhen/ev/tmp/EasyCV/package/easycv/models/builder.py", line 19, in build_backbone
return build(cfg, BACKBONES)
File "/home/zhenzhen/ev/tmp/EasyCV/package/easycv/models/builder.py", line 15, in build
return build_from_cfg(cfg, registry, default_args)
File "/home/zhenzhen/ev/tmp/EasyCV/package/easycv/utils/registry.py", line 80, in build_from_cfg
return obj_cls(**args)
File "/home/zhenzhen/ev/tmp/EasyCV/package/easycv/models/backbones/pytorch_image_models_wrapper.py", line 100, in init
exportable, no_jit, **kwargs)
TypeError: create_model() got multiple values for argument 'pretrained'