Skip to content

GitLab

"tests/test_cli/vscode:/vscode.git/clone" did not exist on "351078f13c5cd929848bb79df17e14f2c6a5eef3"
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):
FLAGS.enable_eager = False
FLAGS.distribution_strategy = 'off'
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()
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