Change hook_cfg type access to hook_cfg.get() (#409)

open-mmlab · Sep 30, 2022 · 3f80791 · 3f80791
1 parent aa3343c
commit 3f80791
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmselfsup/apis/train.py b/mmselfsup/apis/train.py
@@ -170,7 +170,7 @@ def train_model(model,
             assert isinstance(hook_cfg, dict), \
                 'Each item in custom_hooks expects dict type, but got ' \
                 f'{type(hook_cfg)}'
-            if hook_cfg.type == 'DeepClusterHook':
+            if hook_cfg.get('type', None) == 'DeepClusterHook':
                 common_params = dict(dist_mode=True, data_loaders=data_loaders)
             else:
                 common_params = dict()