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

Attempting to resolve the timeout in multi worker test by dropping the steps per epoch from 5 to 2.

Browse Source 
PiperOrigin-RevId: 242722233
This commit is contained in:
Rick Chao 2019-04-09 12:54:54 -07:00 committed by TensorFlower Gardener
parent aefd13f65d
commit f57bac3e79
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/keras/distribute/multi_worker_callback_test.py
View File

@ -61,7 +61,7 @@ def generate_callback_test_function(custom_callable):
self._make_mock_run_std_server()):
strategy = get_strategy_object(strategy_cls)
batch_size = 64
steps = 5
steps = 2
train_ds, _ = _mnist_synthetic_dataset(batch_size, steps)
with strategy.scope():
model = _get_model((28, 28, 1))

6
tensorflow/python/keras/distribute/multi_worker_test.py
View File

@ -250,7 +250,7 @@ def _run_standalone_client(test_obj, strategy, cluster_spec):
def worker_fn(strategy):
with ops.Graph().as_default():
batch_size = 64
steps = 5
steps = 2
with strategy.scope():
train_ds, _ = _mnist_synthetic_dataset(batch_size, steps)
@ -349,7 +349,7 @@ class KerasMultiWorkerTestIndependentWorker(test_base.IndependentWorkerTestBase,
verification_callback.is_between_graph = \
strategy.extended.experimental_between_graph
batch_size = 64
steps = 5
steps = 2
train_ds, _ = _mnist_synthetic_dataset(batch_size, steps)
with strategy.scope():
model = _get_model((28, 28, 1))
@ -404,7 +404,7 @@ class KerasMultiWorkerTestIndependentWorker(test_base.IndependentWorkerTestBase,
with test.mock.patch.object(dc, '_run_std_server',
self._make_mock_run_std_server()):
batch_size = 64
steps = 5
steps = 2
strategy = strategy_cls()
verification_callback.is_between_graph = \
strategy.extended.experimental_between_graph