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

Turn on VariablePolicy for TPUStrategy.

Browse Source 
PiperOrigin-RevId: 337261118
Change-Id: I65e13d10f7766d9d637391e1f4714718f2f6a85c
This commit is contained in:
Ruoxin Sang 2020-10-15 01:30:35 -07:00 committed by TensorFlower Gardener
parent 6b3a88b6e9
commit fbd73cd25d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/distribute/tpu_strategy.py
View File

@ -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)