Skip to content

GitLab

Commit 17ef6405 authored by Taylor Robie's avatar Taylor Robie Committed by Toby Boyd
Browse files

reduce test batch size for 1 GPU no_dist_strat (#6564)

parent e08b6286
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
official/resnet/keras/keras_imagenet_benchmark.py
View file @ 17ef6405
...@@ -186,7 +186,8 @@ class Resnet50KerasBenchmarkBase(keras_benchmark.KerasBenchmark): ...@@ -186,7 +186,8 @@ class Resnet50KerasBenchmarkBase(keras_benchmark.KerasBenchmark):
FLAGS.enable_eager = False FLAGS.enable_eager = False
FLAGS.distribution_strategy = 'off' FLAGS.distribution_strategy = 'off'
FLAGS.model_dir = self._get_model_dir('benchmark_graph_1_gpu_no_dist_strat') FLAGS.model_dir = self._get_model_dir('benchmark_graph_1_gpu_no_dist_strat')
FLAGS.batch_size = 128 FLAGS.batch_size = 96 # BatchNorm is less efficient in legacy graph mode
# due to its reliance on v1 cond.
self._run_and_report_benchmark() self._run_and_report_benchmark()
def benchmark_1_gpu(self): def benchmark_1_gpu(self):
......
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