Skip to content

GitLab

Commit 1025682f authored by Zongwei Zhou's avatar Zongwei Zhou Committed by A. Unique TensorFlower
Browse files

Reduce Resnet50 pure eager FP32 test batch size 

PiperOrigin-RevId: 306287724
parent 1e8c0fc3
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/benchmark/resnet_ctl_imagenet_benchmark.py
View file @ 1025682f
......@@ -288,7 +288,7 @@ class Resnet50CtlBenchmarkBase(CtlBenchmark):
FLAGS.num_gpus = 1
FLAGS.distribution_strategy = 'one_device'
FLAGS.model_dir = self._get_model_dir('benchmark_1_gpu_eager')
FLAGS.batch_size = 128
FLAGS.batch_size = 120
FLAGS.use_tf_function = False
FLAGS.use_tf_while_loop = False
FLAGS.single_l2_loss_op = True
......
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