Skip to content

GitLab

"...io/image/git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "57e6e30203d24d13f361da2ec98fbd8e87a2b4b3"
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): ...@@ -301,7 +301,7 @@ class Resnet50CtlBenchmarkBase(CtlBenchmark):
FLAGS.num_gpus = 1 FLAGS.num_gpus = 1
FLAGS.distribution_strategy = 'one_device' FLAGS.distribution_strategy = 'one_device'
FLAGS.model_dir = self._get_model_dir('benchmark_1_gpu_fp16_eager') FLAGS.model_dir = self._get_model_dir('benchmark_1_gpu_fp16_eager')
FLAGS.batch_size = 250 FLAGS.batch_size = 240
FLAGS.dtype = 'fp16' FLAGS.dtype = 'fp16'
FLAGS.use_tf_function = False FLAGS.use_tf_function = False
FLAGS.use_tf_while_loop = 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