Skip to content

GitLab

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

Increase batch size for Resnet pure eager tests due to recent memory optimizations 

PiperOrigin-RevId: 293498210
parent 4a086ad5
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/benchmark/resnet_ctl_imagenet_benchmark.py
View file @ faf5ad73
...@@ -279,7 +279,7 @@ class Resnet50CtlBenchmarkBase(CtlBenchmark): ...@@ -279,7 +279,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 = 108 FLAGS.batch_size = 128
FLAGS.use_tf_function = False FLAGS.use_tf_function = False
FLAGS.single_l2_loss_op = True FLAGS.single_l2_loss_op = True
self._run_and_report_benchmark() self._run_and_report_benchmark()
...@@ -291,7 +291,7 @@ class Resnet50CtlBenchmarkBase(CtlBenchmark): ...@@ -291,7 +291,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 = 216 FLAGS.batch_size = 250
FLAGS.dtype = 'fp16' FLAGS.dtype = 'fp16'
FLAGS.use_tf_function = False FLAGS.use_tf_function = 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