Skip to content

Commit c2afd05

Browse files
committed
Fixes #2455
1 parent f30358c commit c2afd05

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

pytorch_lightning/callbacks/early_stopping.py

+2
Original file line numberDiff line numberDiff line change
@@ -151,7 +151,9 @@ def _run_early_stopping_check(self, trainer, pl_module):
151151
# check flag across all GPUs
152152
if trainer.use_ddp or trainer.use_ddp2:
153153
should_stop = torch.tensor(int(should_stop), device=pl_module.device)
154+
print(should_stop)
154155
dist.all_reduce(should_stop, op=dist.ReduceOp.MAX)
156+
print(should_stop)
155157

156158
# do actual stop
157159
if should_stop:

0 commit comments

Comments
 (0)