From d6b43ac543256cc28ff3b637968f40880abf8171 Mon Sep 17 00:00:00 2001 From: Scott Zhu Date: Mon, 15 Jul 2019 09:53:46 -0700 Subject: [PATCH] Use the default strategy if no strategy is specified in v2 inside the new V2 codepath. PiperOrigin-RevId: 258183084 --- tensorflow/python/keras/engine/training_v2.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/python/keras/engine/training_v2.py b/tensorflow/python/keras/engine/training_v2.py index 1bb115a8af4..6acf2a4c443 100644 --- a/tensorflow/python/keras/engine/training_v2.py +++ b/tensorflow/python/keras/engine/training_v2.py @@ -371,8 +371,8 @@ class Loop(training_utils.TrainingLoop): def _get_distribution_strategy(model): if model._distribution_strategy: return model._distribution_strategy - # TODO(scottzhu): might want to just get the default strategy in future. - elif distribution_strategy_context.has_strategy(): + elif not distribution_strategy_context.has_strategy(): + # Use the default strategy if no strategy is specified. return distribution_strategy_context.get_strategy() else: return None