Skip to content

Commit cc8d1cd

Browse files
committed
Fixes #2455
1 parent f113088 commit cc8d1cd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytorch_lightning/callbacks/early_stopping.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -152,7 +152,7 @@ def _run_early_stopping_check(self, trainer, pl_module):
152152
should_stop = torch.tensor(int(should_stop), device=pl_module.device)
153153
if trainer.use_ddp or trainer.use_ddp2:
154154
dist.barrier()
155-
dist.all_reduce(should_stop, op=dist.ReduceOp.MAX)
155+
dist.all_reduce(should_stop, op=dist.reduce_op.MAX)
156156

157157
print(f'RANK: {trainer.global_rank} SHOULD STOP: {should_stop} BEST: {self.best_score}')
158158

0 commit comments

Comments
 (0)