Skip to content

GitLab

"...text-generation-inference.git" did not exist on "c1fe28d694757a6a90426a83006292dc76512f66"
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
Hide 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): ...@@ -288,7 +288,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_eager') 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_function = False
FLAGS.use_tf_while_loop = False FLAGS.use_tf_while_loop = False
FLAGS.single_l2_loss_op = True 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