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

Re-enable multi_worker_tutorial_test (by fixing the timeout on shorter epoch/steps) 

PiperOrigin-RevId: 312842049
Change-Id: If0d7f0fcb4463c718f5532f62cca17ac23cab99a
This commit is contained in:
Rick Chao 2020-05-22 10:43:42 -07:00 committed by TensorFlower Gardener
parent 29fb4d12a7
commit c64097cb5f
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -432,7 +432,6 @@ py_test(
tags = [
"noasan", # TODO(b/156029134)
"nomsan", # TODO(b/156029134)
"notap", # TODO(b/157253858)
"notsan", # TODO(b/156029134)
],
deps = [

4
tensorflow/python/keras/distribute/multi_worker_tutorial_test.py
View File

@ -120,8 +120,8 @@ class MultiWorkerTutorialTest(parameterized.TestCase, test.TestCase):
multi_worker_model.fit(
multi_worker_dataset,
epochs=3,
steps_per_epoch=70,
epochs=2,
steps_per_epoch=20,
callbacks=callbacks)
with test_util.skip_if_error(self, errors_impl.UnavailableError):