Fixes #380

2019-10-18 15:33:00 +02:00 · 2019-10-18 15:33:00 +02:00 · 416187fd56
parent 3dfcef6994
commit 416187fd56
1 changed files with 1 additions and 1 deletions
--- 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