Skip to content

GitLab

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

Enable single l2 loss op in Resnet50 CTL 1gpu eager benchmark 

PiperOrigin-RevId: 264937518
parent 39aea35f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
official/resnet/ctl/ctl_imagenet_benchmark.py
View file @ 211bf1c0
......@@ -215,6 +215,7 @@ class Resnet50CtlBenchmarkBase(CtlBenchmark):
FLAGS.model_dir = self._get_model_dir('benchmark_1_gpu_eager')
FLAGS.batch_size = 64
FLAGS.use_tf_function = False
FLAGS.single_l2_loss_op = True
self._run_and_report_benchmark()
def benchmark_8_gpu(self):
......
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