diff options
author | 2016-06-09 13:56:54 -0800 | |
---|---|---|
committer | 2016-06-09 15:03:40 -0700 | |
commit | 48f2522176f0e2b3d30304b247658c240d81fc88 (patch) | |
tree | 11b39962af5f062ed712244556608f0212c506a8 /tensorflow/python/training/training.py | |
parent | 1fde2817529524f8b0d59ab24400ca8c2675fcf7 (diff) |
Add basic_train_loop() as an example for higher level frameworks to copy or
reuse. It can also be used directly for simple training.
Fix Coordinator.clear_stop() to also clear the exception to raise. Add test.
Add SummaryWriter.reopen(), with tests. This is needed to properly handle
summaries when create a session more than once in a Supervior.
In Supervisor.prepare_or_wait_for_session() reopen the summary writer.
At then end of Supervisor.managed_session() correctly close the summary write
and clear the running threads even if an exception was reported.
Change: 124500982
Diffstat (limited to 'tensorflow/python/training/training.py')
-rw-r--r-- | tensorflow/python/training/training.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/training/training.py b/tensorflow/python/training/training.py index 84ffd6c1da..82a955276b 100644 --- a/tensorflow/python/training/training.py +++ b/tensorflow/python/training/training.py @@ -173,6 +173,7 @@ from tensorflow.python.training.queue_runner import * from tensorflow.python.training import input as _input from tensorflow.python.training.input import * +from tensorflow.python.training.basic_loops import basic_train_loop from tensorflow.python.training.device_setter import replica_device_setter from tensorflow.python.training.saver import generate_checkpoint_state_proto from tensorflow.python.training.saver import get_checkpoint_state |