diff options
author | 2018-02-05 15:30:54 -0800 | |
---|---|---|
committer | 2018-02-05 15:34:40 -0800 | |
commit | 395550bc423f6a5d9c96233f34f21fce6bd23b4e (patch) | |
tree | 7c43f68e8a3d005328083cdf414bad8dc667f72f /tensorflow/contrib/slim | |
parent | 2074a568810ea20c267e9d063a066b81ad491eed (diff) |
Assign total_loss in order not to crash if training loop exists early.
PiperOrigin-RevId: 184596877
Diffstat (limited to 'tensorflow/contrib/slim')
-rw-r--r-- | tensorflow/contrib/slim/python/slim/learning.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/slim/python/slim/learning.py b/tensorflow/contrib/slim/python/slim/learning.py index 54362c87b5..83f33806e0 100644 --- a/tensorflow/contrib/slim/python/slim/learning.py +++ b/tensorflow/contrib/slim/python/slim/learning.py @@ -738,6 +738,7 @@ def train(train_op, if summary_writer is not None: train_step_kwargs['summary_writer'] = sv.summary_writer + total_loss = 0 should_retry = True while should_retry: try: |