[Fix] Fix hub.get_model failure on some mmcls models (#784)

open-mmlab · Dec 4, 2022 · 5ea21f2 · 5ea21f2
1 parent 7e787d8
commit 5ea21f2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mmengine/hub/hub.py b/mmengine/hub/hub.py
@@ -75,6 +75,8 @@ def get_model(cfg_path: str, pretrained: bool = False, **kwargs):
     package = cfg_path.split('::')[0]
     with DefaultScope.overwrite_default_scope(package):  # type: ignore
         cfg = get_config(cfg_path, pretrained)
+        if 'data_preprocessor' in cfg:
+            cfg.model.data_preprocessor = cfg.data_preprocessor
         models_module = importlib.import_module(f'{package}.utils')
         models_module.register_all_modules()  # type: ignore
         model = MODELS.build(cfg.model, default_args=kwargs)