wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix for #2051, always init with -inf

Browse Source 
The comparison `self.monitor_op == torch.gt` was always false, compare with mode instead for #2051
This commit is contained in:
Mike Clark
2020-06-03 13:22:58 +08:00
committed by GitHub
parent 8211256c46
commit 477b2549de
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/callbacks/early_stopping.py
+1 -1
View File
@@ -75,7 +75,7 @@ class EarlyStopping(Callback):
if self.verbose > 0:
log.info(f'EarlyStopping mode set to {self.mode} for monitoring {self.monitor}.')
self.min_delta *= 1 if self.monitor_op == torch.gt else -1
self.min_delta *= 1 if self.mode == 'min' else -1
def _validate_condition_metric(self, logs):
"""