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

Fix flaky loss_scale_optimizer_test.

Browse Source 
The test would timeout about 2% of the time.

PiperOrigin-RevId: 308914241
Change-Id: Ic24402ef3f87152f94b96eddf76bc2372d7a5d27
This commit is contained in:
Reed Wanderman-Milne 2020-04-28 15:54:18 -07:00 committed by TensorFlower Gardener
parent 1ce189fed3
commit 86a6005136
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/keras/mixed_precision/experimental/BUILD
View File

@ -180,7 +180,7 @@ py_library(
cuda_py_test( cuda_py_test(
name = "loss_scale_optimizer_test", name = "loss_scale_optimizer_test",
size = "small", size = "medium",
srcs = ["loss_scale_optimizer_test.py"], srcs = ["loss_scale_optimizer_test.py"],
python_version = "PY3", python_version = "PY3",
deps = [ deps = [