From bf6cc418dcfcee1db8a76586f3c47d561cacd0fa Mon Sep 17 00:00:00 2001 From: wutongshenqiu <44188071+wutongshenqiu@users.noreply.github.com> Date: Sun, 27 Mar 2022 17:41:13 +0800 Subject: [PATCH] use get_codebase_config --- mmdeploy/codebase/base/task.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/mmdeploy/codebase/base/task.py b/mmdeploy/codebase/base/task.py index bc77f712ce..330a61c0a2 100644 --- a/mmdeploy/codebase/base/task.py +++ b/mmdeploy/codebase/base/task.py @@ -7,7 +7,8 @@ import torch from torch.utils.data import DataLoader, Dataset -from mmdeploy.utils import get_backend_config, get_codebase, get_root_logger +from mmdeploy.utils import (get_backend_config, get_codebase, + get_codebase_config, get_root_logger) from mmdeploy.utils.dataset import is_can_sort_dataset, sort_dataset @@ -295,8 +296,8 @@ def from_mmrazor(self) -> bool: Raises: TypeError: An error when type of `from_mmrazor` is not boolean. """ - from_mmrazor = getattr(self.deploy_cfg.codebase_config, 'from_mmrazor', - False) + codebase_config = get_codebase_config(self.deploy_cfg) + from_mmrazor = codebase_config.get('from_mmrazor', False) if not isinstance(from_mmrazor, bool): raise TypeError('`from_mmrazor` attribute must be boolean type! ' f'but got: {from_mmrazor}')