From 86a600513679a671d041cbf414b44a3445ed877f Mon Sep 17 00:00:00 2001 From: Reed Wanderman-Milne Date: Tue, 28 Apr 2020 15:54:18 -0700 Subject: [PATCH] Fix flaky loss_scale_optimizer_test. The test would timeout about 2% of the time. PiperOrigin-RevId: 308914241 Change-Id: Ic24402ef3f87152f94b96eddf76bc2372d7a5d27 --- tensorflow/python/keras/mixed_precision/experimental/BUILD | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/python/keras/mixed_precision/experimental/BUILD b/tensorflow/python/keras/mixed_precision/experimental/BUILD index 18390b49008..9462051cb83 100644 --- a/tensorflow/python/keras/mixed_precision/experimental/BUILD +++ b/tensorflow/python/keras/mixed_precision/experimental/BUILD @@ -180,7 +180,7 @@ py_library( cuda_py_test( name = "loss_scale_optimizer_test", - size = "small", + size = "medium", srcs = ["loss_scale_optimizer_test.py"], python_version = "PY3", deps = [