Fix Keras Trivial Model 1 GPU test (#6481)

8e7051a8 · Haoyu Zhang · Toby Boyd · 6925d23b · 8e7051a8
Commit 8e7051a8 authored Mar 29, 2019 by Haoyu Zhang Committed by Toby Boyd Mar 29, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/resnet/keras/keras_imagenet_benchmark.py official/resnet/keras/keras_imagenet_benchmark.py +1 -1

No files found.
--- a/official/resnet/keras/keras_imagenet_benchmark.py
+++ b/official/resnet/keras/keras_imagenet_benchmark.py
@@ -463,7 +463,7 @@ class TrivialKerasBenchmark(keras_benchmark.KerasBenchmark):
    """Test trivial Keras model (input pipeline) with 1 GPU."""
    self._setup()

-    FLAGS.num_gpus = 8
+    FLAGS.num_gpus = 1
    FLAGS.enable_eager = False
    FLAGS.model_dir = self._get_model_dir('benchmark_graph_1_gpu')
    FLAGS.batch_size = 256