From 1f25122749acd2c2dffc2ee111c35056b25ef864 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Thu, 4 Apr 2019 17:25:34 +0200 Subject: [PATCH] Removed reference to early stopping from docstring Signed-off-by: Jim Martens --- src/twomartens/masterthesis/aae/train.py | 10 ---------- 1 file changed, 10 deletions(-) diff --git a/src/twomartens/masterthesis/aae/train.py b/src/twomartens/masterthesis/aae/train.py index 81eb1bd..fa4af27 100644 --- a/src/twomartens/masterthesis/aae/train.py +++ b/src/twomartens/masterthesis/aae/train.py @@ -71,16 +71,6 @@ def train_simple(dataset: tf.data.Dataset, lr: initial learning rate (default: 0.002) train_epoch: number of epochs to train (default: 80) verbose: if True prints train progress info to console (default: True) - - Notes: - The training stops early if for ``GRACE`` number of epochs the loss is not - decreasing. Specifically all individual losses are accounted for and any one - of those not decreasing triggers a ``strike``. If the total loss, which is - a sum of all individual losses, is also not decreasing and has a total - value of more than ``TOTAL_LOSS_GRACE_CAP``, the counter for the remaining grace period is - decreased. If in any epoch afterwards all losses are decreasing the grace - period is reset to ``GRACE``. Lastly the training loop will be stopped early - if the grace counter reaches ``0`` at the end of an epoch. """ # non-preserved tensors