Register DistSamplerSeedHook for EpochBasedRunner only

Co-authored-by: Ma Zerun <mzr1996@163.com>
open-mmlab · Oct 27, 2021 · 2169cb5 · 2169cb5
1 parent 94b6609
commit 2169cb5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmcls/apis/train.py b/mmcls/apis/train.py
@@ -135,7 +135,7 @@ def train_model(model,
         cfg.log_config,
         cfg.get('momentum_config', None),
         custom_hooks_config=cfg.get('custom_hooks', None))
-    if distributed and cfg.runner['type'] != 'IterBasedRunner':
+    if distributed and cfg.runner['type'] == 'EpochBasedRunner':
         runner.register_hook(DistSamplerSeedHook())
 
     # register eval hooks