From 17537b944b27795c6193a99b254744c0998c44aa Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Thu, 4 Apr 2019 17:26:56 +0200 Subject: [PATCH] Removed left-over reference to early stopping Signed-off-by: Jim Martens --- src/twomartens/masterthesis/aae/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/aae/train.py b/src/twomartens/masterthesis/aae/train.py index cd9183c..80c402d 100644 --- a/src/twomartens/masterthesis/aae/train.py +++ b/src/twomartens/masterthesis/aae/train.py @@ -55,7 +55,7 @@ def train_simple(dataset: tf.data.Dataset, """ Trains auto-encoder for given data set. - This function provides early stopping and creates checkpoints after every + This function creates checkpoints after every epoch as well as after finishing training (or stopping early). When starting this function with the same ``iteration`` then the training will try to continue where it ended last time by restoring a saved checkpoint.