From 5ea21f2424830871401d5a978a48815b46cc17af Mon Sep 17 00:00:00 2001
From: Qian Zhao <112053249+C1rN09@users.noreply.github.com>
Date: Sun, 4 Dec 2022 10:40:48 +0800
Subject: [PATCH] [Fix] Fix hub.get_model failure on some mmcls models (#784)

---
 mmengine/hub/hub.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/mmengine/hub/hub.py b/mmengine/hub/hub.py
index b7b0d4ade4..b24ac2c125 100644
--- 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)