remove DistSamplerSeedHook for IterBasedRunner

open-mmlab · Oct 27, 2021 · 94b6609 · 94b6609
1 parent 52e6256
commit 94b6609
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:
+    if distributed and cfg.runner['type'] != 'IterBasedRunner':
         runner.register_hook(DistSamplerSeedHook())
 
     # register eval hooks