[Fix] fix potential bug (#647)

open-mmlab · Dec 27, 2022 · f73306d · f73306d
1 parent d731546
commit f73306d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmselfsup/apis/train.py b/mmselfsup/apis/train.py
@@ -158,7 +158,7 @@ def train_model(model,
     # register hooks
     runner.register_training_hooks(cfg.lr_config, optimizer_config,
                                    cfg.checkpoint_config, cfg.log_config)
-    if distributed:
+    if distributed and cfg.runner.type == 'EpochBasedRunner':
         runner.register_hook(DistSamplerSeedHook())
 
     # register custom hooks