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

Temporarily change the size of keras/distribute/multi_worker_test as it's frequently timing out.

Browse Source 
PiperOrigin-RevId: 242542350
This commit is contained in:
Rick Chao 2019-04-08 14:42:50 -07:00 committed by TensorFlower Gardener
parent 73ae2a64c7
commit 115112f7fb
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/keras/distribute/BUILD
View File

@ -279,6 +279,8 @@ distribute_py_test(
cuda_py_test( cuda_py_test(
name = "multi_worker_test", name = "multi_worker_test",
# TODO(rchao): Investigate the timeout issue and size it back to medium.
size = "large",
srcs = ["multi_worker_test.py"], srcs = ["multi_worker_test.py"],
additional_deps = [":multi_worker_test_main_lib"], additional_deps = [":multi_worker_test_main_lib"],
shard_count = 32, shard_count = 32,