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

Use the default strategy if no strategy is specified in v2 inside the new V2 codepath.

Browse Source 
PiperOrigin-RevId: 258183084
This commit is contained in:
Scott Zhu 2019-07-15 09:53:46 -07:00 committed by TensorFlower Gardener
parent 8df7ce02ba
commit d6b43ac543
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/python/keras/engine/training_v2.py
View File

@ -371,8 +371,8 @@ class Loop(training_utils.TrainingLoop):
def _get_distribution_strategy(model): def _get_distribution_strategy(model):
if model._distribution_strategy: if model._distribution_strategy:
return model._distribution_strategy return model._distribution_strategy
# TODO(scottzhu): might want to just get the default strategy in future. elif not distribution_strategy_context.has_strategy():
elif distribution_strategy_context.has_strategy(): # Use the default strategy if no strategy is specified.
return distribution_strategy_context.get_strategy() return distribution_strategy_context.get_strategy()
else: else:
return None return None