Skip to content

GitLab

Commit 8e7051a8 authored by Haoyu Zhang's avatar Haoyu Zhang Committed by Toby Boyd
Browse files

Fix Keras Trivial Model 1 GPU test (#6481)

parent 6925d23b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/resnet/keras/keras_imagenet_benchmark.py
View file @ 8e7051a8
...@@ -463,7 +463,7 @@ class TrivialKerasBenchmark(keras_benchmark.KerasBenchmark): ...@@ -463,7 +463,7 @@ class TrivialKerasBenchmark(keras_benchmark.KerasBenchmark):
"""Test trivial Keras model (input pipeline) with 1 GPU.""" """Test trivial Keras model (input pipeline) with 1 GPU."""
self._setup() self._setup()
FLAGS.num_gpus = 8 FLAGS.num_gpus = 1
FLAGS.enable_eager = False FLAGS.enable_eager = False
FLAGS.model_dir = self._get_model_dir('benchmark_graph_1_gpu') FLAGS.model_dir = self._get_model_dir('benchmark_graph_1_gpu')
FLAGS.batch_size = 256 FLAGS.batch_size = 256
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment