From 6a33f0d4833bb0b6b1b65d019eac1bf873e51dc1 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 15 Jul 2019 14:54:38 -0400 Subject: [PATCH] made early stop checkpoint optional --- pytorch_lightning/models/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index a4b02bb9..89b6d5ce 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -45,7 +45,7 @@ class Trainer(TrainerIO): check_val_every_n_epoch=1, fast_dev_run=False, accumulate_grad_batches=1, - enable_early_stop=True, max_nb_epochs=1000, min_nb_epochs=1, + max_nb_epochs=1000, min_nb_epochs=1, train_percent_check=1.0, val_percent_check=1.0, test_percent_check=1.0, val_check_interval=0.95, log_save_interval=100, add_log_row_interval=10, lr_scheduler_milestones=None,