From 416187fd5671c6f8e8562fdc374c0603cfe6e13c Mon Sep 17 00:00:00 2001 From: William Falcon Date: Fri, 18 Oct 2019 15:33:00 +0200 Subject: [PATCH] Fixes #380 --- pytorch_lightning/callbacks/pt_callbacks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/callbacks/pt_callbacks.py b/pytorch_lightning/callbacks/pt_callbacks.py index e0aa2e6d34..12d8919cb2 100644 --- a/pytorch_lightning/callbacks/pt_callbacks.py +++ b/pytorch_lightning/callbacks/pt_callbacks.py @@ -175,7 +175,7 @@ class ModelCheckpoint(Callback): """ def __init__(self, filepath, monitor='val_loss', verbose=0, - save_best_only=False, save_weights_only=False, + save_best_only=True, save_weights_only=False, mode='auto', period=1, prefix=''): super(ModelCheckpoint, self).__init__() self.monitor = monitor