use get_codebase_config

open-mmlab · Mar 27, 2022 · bf6cc41 · bf6cc41
1 parent 934dbc6
commit bf6cc41
Showing 1 changed file with 4 additions and 3 deletions.
diff --git 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}')