From fbd73cd25d57a67687f761b894007ee90d4bda5d Mon Sep 17 00:00:00 2001 From: Ruoxin Sang Date: Thu, 15 Oct 2020 01:30:35 -0700 Subject: [PATCH] Turn on VariablePolicy for TPUStrategy. PiperOrigin-RevId: 337261118 Change-Id: I65e13d10f7766d9d637391e1f4714718f2f6a85c --- tensorflow/python/distribute/tpu_strategy.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/python/distribute/tpu_strategy.py b/tensorflow/python/distribute/tpu_strategy.py index 719147db5ee..d8eabf64e5d 100644 --- a/tensorflow/python/distribute/tpu_strategy.py +++ b/tensorflow/python/distribute/tpu_strategy.py @@ -740,7 +740,7 @@ class TPUExtended(distribute_lib.StrategyExtendedV1): atexit.register(async_wait) # Flag to turn on VariablePolicy - self._use_var_policy = True + self._use_var_policy = False def _validate_colocate_with_variable(self, colocate_with_variable): distribute_utils. validate_colocate(colocate_with_variable, self)