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 early stopping off by 2 (min_epochs) (#617)

Browse Source 
* fix early stopping off by 2

* add min_epochs example in docs
This commit is contained in:
Adrian Wälchli
2019-12-09 10:32:49 -08:00
committed by William Falcon
parent d562172b4c
commit e2ee4ddbdb
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/trainer/training_loop.py
+4 -1
View File
@@ -50,6 +50,9 @@ To modify this behavior, pass in your own EarlyStopping callback.
# pass in your own to override the default callback
trainer = Trainer(early_stop_callback=early_stop_callback)
# pass in min_epochs to enable the callback after min_epochs have run
trainer = Trainer(early_stop_callback=early_stop_callback, min_epochs=5)
# pass in None to disable it
trainer = Trainer(early_stop_callback=None)
@@ -339,7 +342,7 @@ class TrainerTrainLoopMixin(ABC):
self.reduce_lr_on_plateau_scheduler.step(val_loss, epoch=self.current_epoch)
# early stopping
met_min_epochs = epoch > self.min_epochs
met_min_epochs = epoch >= self.min_epochs - 1
if self.enable_early_stop and (met_min_epochs or self.fast_dev_run):
should_stop = self.early_stop_callback.on_epoch_end(epoch=epoch,
logs=self.callback_metrics)