diff --git a/mmengine/hub/hub.py b/mmengine/hub/hub.py index b7b0d4ade4c56f7c23497549ad9c8f4ae6e64009..b24ac2c125143789c8553086eec7767b9c761c41 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)