cleaning up demos

2026-09-11 12:31:23 +08:00 · 2019-10-05 15:52:52 -04:00
parent 7b982f2280
commit 18f58d4bf4
1 changed files with 1 additions and 1 deletions
@@ -704,7 +704,7 @@ class Trainer(TrainerIO):
                        warnings.warn(msg)
                        break

-        if self.nb_gpu_nodes > 1 and (self.use_ddp or self.use_ddp2):
+        if self.use_ddp:
            # wait for all processes to catch up
            dist.barrier()