experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix keras_multi_worker test failure.

Browse Source 
PiperOrigin-RevId: 223422542
This commit is contained in:
Yuefeng Zhou 2018-11-29 15:28:37 -08:00 committed by TensorFlower Gardener
parent 3a9b2ae05d
commit bbad7c0a07
1 changed files with 1 additions and 1 deletions

2
tensorflow/python/keras/engine/distributed_training_utils.py
View File

@ -328,7 +328,7 @@ def init_restore_or_wait_for_variables():
"""Initialize or restore variables or wait for variables to be initialized."""
session = K._get_session() # pylint: disable=protected-access
worker_context = dc_context.get_current_worker_context()
if not worker_context or worker_context.should_init:
if not worker_context or worker_context.experimental_should_init:
# TODO(yuefengz): if checkpoints exit, restore from checkpoint.
K._initialize_variables(session) # pylint: disable=protected-access
else: