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

Enable multi_worker_test with "multi_and_single_gpu" tag.

Browse Source 
PiperOrigin-RevId: 290397331
Change-Id: Id88083fff97a0bed45829952d10812cd0ddfa91b
This commit is contained in:
Yanhui Liang 2020-01-17 22:34:12 -08:00 committed by TensorFlower Gardener
parent 96170c0207
commit e87fbf3b6c
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/distribute/BUILD
View File

@ -320,8 +320,8 @@ cuda_py_test(
srcs = ["multi_worker_test.py"], srcs = ["multi_worker_test.py"],
shard_count = 32, shard_count = 32,
tags = [ tags = [
"multi_and_single_gpu",
"no_oss", # TODO(b/130369494): Investigate why it times out on OSS. "no_oss", # TODO(b/130369494): Investigate why it times out on OSS.
# TODO(b/123307453): Add "multi_and_single_gpu",
], ],
deps = [ deps = [
":multi_worker_testing_utils", ":multi_worker_testing_utils",