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

Remove TODO since workaround makes sense.

Browse Source 
PiperOrigin-RevId: 297535246
Change-Id: I4bb3006bd42b0d637d6517049b7e6f16c4f4e63b
This commit is contained in:
Ken Franko 2020-02-27 00:54:08 -08:00 committed by TensorFlower Gardener
parent a0434c8fa6
commit dac3721584
1 changed files with 3 additions and 2 deletions

5
tensorflow/examples/saved_model/integration_tests/distribution_strategy_utils.py
View File

@ -32,8 +32,9 @@ _strategies = [
strategy_combinations.tpu_strategy,
]
# TODO(b/145386854): The presence of GPU strategies upsets TPU initialization,
# despite their test instances being skipped early on.
# The presence of GPU strategies upsets TPU initialization,
# despite their test instances being skipped early. This is a workaround
# for b/145386854.
if "test_tpu" in sys.argv[0]:
_strategies = [s for s in _strategies if "GPU" not in str(s)]