From 115112f7fb53d31fbfbbe75d761e993abc5c8c26 Mon Sep 17 00:00:00 2001 From: Rick Chao Date: Mon, 8 Apr 2019 14:42:50 -0700 Subject: [PATCH] Temporarily change the size of keras/distribute/multi_worker_test as it's frequently timing out. PiperOrigin-RevId: 242542350 --- tensorflow/python/keras/distribute/BUILD | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tensorflow/python/keras/distribute/BUILD b/tensorflow/python/keras/distribute/BUILD index 025e458eadd..ac32453d714 100644 --- a/tensorflow/python/keras/distribute/BUILD +++ b/tensorflow/python/keras/distribute/BUILD @@ -279,6 +279,8 @@ distribute_py_test( cuda_py_test( name = "multi_worker_test", + # TODO(rchao): Investigate the timeout issue and size it back to medium. + size = "large", srcs = ["multi_worker_test.py"], additional_deps = [":multi_worker_test_main_lib"], shard_count = 32,