diff options
| author |  Yuefeng Zhou <yuefengz@google.com> | 2018-08-24 15:19:16 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-24 15:23:12 -0700 |
| commit | d8dc2a1cd23db7a6b15083f14042bab272756ae2 (patch) | |
| tree | b31b9821b579abe80272ab9bc867290603f3f64a /tensorflow/contrib/distribute/python/mirrored_strategy_multigpu_test.py | |
| parent | 28e4880df833f443711ced08d8bc3c64f87f4b44 (diff) | |
Reduce learning_rate multi-worker MirroredStrategy test.
PiperOrigin-RevId: 210165808
Diffstat (limited to 'tensorflow/contrib/distribute/python/mirrored_strategy_multigpu_test.py')
| -rw-r--r-- | tensorflow/contrib/distribute/python/mirrored_strategy_multigpu_test.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/contrib/distribute/python/mirrored_strategy_multigpu_test.py b/tensorflow/contrib/distribute/python/mirrored_strategy_multigpu_test.py index ac2697958d..a12ff662db 100644 --- a/tensorflow/contrib/distribute/python/mirrored_strategy_multigpu_test.py +++ b/tensorflow/contrib/distribute/python/mirrored_strategy_multigpu_test.py @@ -1271,7 +1271,8 @@ class MultiWorkerMirroredStrategyTest( return strategy def testMinimizeLossGraph(self): - self._test_minimize_loss_graph(self._get_distribution_strategy()) + self._test_minimize_loss_graph(self._get_distribution_strategy(), + learning_rate=0.05) class MultiWorkerMirroredStrategyTestWithChief( @@ -1288,7 +1289,7 @@ class MultiWorkerMirroredStrategyTestWithChief( def testMinimizeLossGraph(self): strategy = mirrored_strategy.MirroredStrategy(num_gpus=context.num_gpus()) strategy.configure(cluster_spec=self._cluster_spec) - self._test_minimize_loss_graph(strategy) + self._test_minimize_loss_graph(strategy, learning_rate=0.05) if __name__ == "__main__": |