From 3a93aaf9e2bf2a5ab80a4dc46afd59c518b3b85f Mon Sep 17 00:00:00 2001 From: Shunta Komatsu <59395084+skmatz@users.noreply.github.com> Date: Fri, 27 Mar 2020 21:42:52 +0900 Subject: [PATCH] Fix outdated docs (#1227) --- pytorch_lightning/trainer/training_io.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/trainer/training_io.py b/pytorch_lightning/trainer/training_io.py index 7d0f22def1..99264b7b5b 100644 --- a/pytorch_lightning/trainer/training_io.py +++ b/pytorch_lightning/trainer/training_io.py @@ -17,7 +17,7 @@ To modify the behavior of checkpointing pass in your own callback. # DEFAULTS used by the Trainer checkpoint_callback = ModelCheckpoint( filepath=os.getcwd(), - save_best_only=True, + save_top_k=1, verbose=True, monitor='val_loss', mode='min',