Make 2 gpu tests num_gpu=2 (was wrong before) (#6187)

25b10ebe · Toby Boyd · Dong Lin · a1ee97e6 · 25b10ebe
Commit 25b10ebe authored Feb 11, 2019 by Toby Boyd Committed by Dong Lin Feb 11, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/resnet/estimator_cifar_benchmark.py official/resnet/estimator_cifar_benchmark.py +1 -1

No files found.
--- a/official/resnet/estimator_cifar_benchmark.py
+++ b/official/resnet/estimator_cifar_benchmark.py
@@ -66,7 +66,7 @@ class EstimatorCifar10BenchmarkTests(tf.test.Benchmark):
  def resnet56_2_gpu(self):
    """Test layers model with Estimator and dist_strat. 2 GPUs."""
    self._setup()
-    flags.FLAGS.num_gpus = 1
+    flags.FLAGS.num_gpus = 2
    flags.FLAGS.data_dir = DATA_DIR
    flags.FLAGS.batch_size = 128
    flags.FLAGS.train_epochs = 182