cleaning up demos

2026-09-11 12:31:23 +08:00 · 2019-10-05 16:01:27 -04:00
parent 56179bdbbb
commit d6c79644af
1 changed files with 1 additions and 1 deletions
@@ -704,7 +704,7 @@ class Trainer(TrainerIO):
                        warnings.warn(msg)
                        break

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