mirrors/mmselfsup
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmselfsup.git synced 2025-06-03 14:59:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Lőrincz-Molnár Szabolcs-Botond 2022-08-09 10:53:12 +03:00 committed by Yixiao Fang
parent e826153b2d
commit a0087327da
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mmselfsup/apis/train.py
View File

@ -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()