-
Notifications
You must be signed in to change notification settings - Fork 9
Description
感谢您的工作,我在尝试跑通您的代码,但是在我使用自己预训练的模型进行微调时,遇到了上述几个库之间版本不兼容的报错,后续查阅资料,下载了兼容的版本:
mmcv 2.0.0rc4 https://github.com/open-mmlab/mmcv
mmdet 3.0.0rc5 https://github.com/open-mmlab/mmdetection
mmengine 0.10.5 https://github.com/open-mmlab/mmengine
mmocr 1.0.0 /home/gx/Union14M-main/mmocr-dev-1.x
但出现了新的报错如下:
Traceback (most recent call last):
File "tools/train.py", line 113, in
main()
File "tools/train.py", line 102, in main
runner = Runner.from_cfg(cfg)
File "/home/gx/anaconda3/envs/maee/lib/python3.8/site-packages/mmengine/runner/runner.py", line 462, in from_cfg
runner = cls(
File "/home/gx/anaconda3/envs/maee/lib/python3.8/site-packages/mmengine/runner/runner.py", line 429, in init
self.model = self.build_model(model)
File "/home/gx/anaconda3/envs/maee/lib/python3.8/site-packages/mmengine/runner/runner.py", line 836, in build_model
model = MODELS.build(model)
File "/home/gx/anaconda3/envs/maee/lib/python3.8/site-packages/mmengine/registry/registry.py", line 570, in build
return self.build_func(cfg, *args, **kwargs, registry=self)
File "/home/gx/anaconda3/envs/maee/lib/python3.8/site-packages/mmengine/registry/build_functions.py", line 232, in build_model_from_cfg
return build_from_cfg(cfg, registry, default_args)
File "/home/gx/anaconda3/envs/maee/lib/python3.8/site-packages/mmengine/registry/build_functions.py", line 121, in build_from_cfg
obj = obj_cls(**args) # type: ignore
File "/home/gx/Union14M-main/mmocr-dev-1.x/mmocr/models/textrecog/recognizers/encoder_decoder_recognizer.py", line 52, in init
self.backbone = MODELS.build(backbone)
File "/home/gx/anaconda3/envs/maee/lib/python3.8/site-packages/mmengine/registry/registry.py", line 570, in build
return self.build_func(cfg, *args, **kwargs, registry=self)
File "/home/gx/anaconda3/envs/maee/lib/python3.8/site-packages/mmengine/registry/build_functions.py", line 232, in build_model_from_cfg
return build_from_cfg(cfg, registry, default_args)
File "/home/gx/anaconda3/envs/maee/lib/python3.8/site-packages/mmengine/registry/build_functions.py", line 121, in build_from_cfg
obj = obj_cls(**args) # type: ignore
File "/home/gx/Union14M-main/mmocr-dev-1.x/mmocr/models/common/backbones/vit.py", line 57, in init
super(VisionTransformer, self).init(
TypeError: init() got an unexpected keyword argument 'vit_config'
查阅资料没有找到有效的解决方式,我猜测仍然是mm库之间的版本问题导致,所以想请问您训练时具体的版本号,谢谢