Reduce Resnet50 pure eager FP32 test batch size

PiperOrigin-RevId: 306287724

Reduce Resnet50 pure eager FP32 test batch size
PiperOrigin-RevId: 306287724
1025682f · Zongwei Zhou · A. Unique TensorFlower · 1e8c0fc3 · 1025682f
Commit 1025682f authored Apr 13, 2020 by Zongwei Zhou Committed by A. Unique TensorFlower Apr 13, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/benchmark/resnet_ctl_imagenet_benchmark.py official/benchmark/resnet_ctl_imagenet_benchmark.py +1 -1

No files found.
--- a/official/benchmark/resnet_ctl_imagenet_benchmark.py
+++ b/official/benchmark/resnet_ctl_imagenet_benchmark.py
@@ -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