From bc99311646a92a1099021ea767c489e033fa61ed Mon Sep 17 00:00:00 2001
From: failure-to-thrive <41901895+failure-to-thrive@users.noreply.github.com>
Date: Wed, 29 Jan 2020 22:03:05 +0300
Subject: [PATCH] Update time_stopping.py

---
 tensorflow_addons/callbacks/time_stopping.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tensorflow_addons/callbacks/time_stopping.py b/tensorflow_addons/callbacks/time_stopping.py
index 9d98f8631a..9734c2cc7e 100644
--- a/tensorflow_addons/callbacks/time_stopping.py
+++ b/tensorflow_addons/callbacks/time_stopping.py
@@ -36,6 +36,7 @@ def __init__(self, seconds=86400, verbose=0):
 
         self.seconds = seconds
         self.verbose = verbose
+        self.stopped_epoch = None
 
     def on_train_begin(self, logs=None):
         self.stopping_time = time.time() + self.seconds
@@ -46,7 +47,7 @@ def on_epoch_end(self, epoch, logs={}):
             self.stopped_epoch = epoch
 
     def on_train_end(self, logs=None):
-        if self.verbose > 0:
+        if self.stopped_epoch is not None and self.verbose > 0:
             formatted_time = datetime.timedelta(seconds=self.seconds)
             msg = "Timed stopping at epoch {} after training for {}".format(
                 self.stopped_epoch + 1, formatted_time