Skip to content

GitLab

Commit 7194a5cc authored by Zongwei Zhou's avatar Zongwei Zhou Committed by A. Unique TensorFlower
Browse files

Reduce Resnet50 pure eager FP16 test batch size 

PiperOrigin-RevId: 306064486
parent 242229af
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/benchmark/resnet_ctl_imagenet_benchmark.py
View file @ 7194a5cc
......@@ -301,7 +301,7 @@ class Resnet50CtlBenchmarkBase(CtlBenchmark):
FLAGS.num_gpus = 1
FLAGS.distribution_strategy = 'one_device'
FLAGS.model_dir = self._get_model_dir('benchmark_1_gpu_fp16_eager')
FLAGS.batch_size = 250
FLAGS.batch_size = 240
FLAGS.dtype = 'fp16'
FLAGS.use_tf_function = False
FLAGS.use_tf_while_loop = False
......
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