diff options
| author |  Jonathan Hseu <jhseu@google.com> | 2018-10-01 15:08:25 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-01 15:11:38 -0700 |
| commit | 9084e999b3caf65833f9651c72bc09eb3094eba5 (patch) | |
| tree | b1f819fea7d867d63b894ba468b72ae04cd53a0c /tensorflow/python/training | |
| parent | df7221d84988e5f7c1cc2775d8f5f44ffdd5918b (diff) | |
Don't run initialize ops if it's empty. Fixes a bug when using the profiler.
PiperOrigin-RevId: 215287936
Diffstat (limited to 'tensorflow/python/training')
| -rw-r--r-- | tensorflow/python/training/session_manager.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/python/training/session_manager.py b/tensorflow/python/training/session_manager.py index 5e4749f306..cd313c2ce0 100644 --- a/tensorflow/python/training/session_manager.py +++ b/tensorflow/python/training/session_manager.py @@ -184,9 +184,11 @@ class SessionManager(object): self._target = master sess = session.Session(self._target, graph=self._graph, config=config) # TODO(jhseu): Delete once tpu.initialize_system() goes away. - sess.run( + initialize_ops = ( distribution_strategy_context.get_distribution_strategy().initialize() ) + if initialize_ops: + sess.run(initialize_ops) if checkpoint_dir and checkpoint_filename_with_path: raise ValueError("Can not provide both checkpoint_dir and " |