Commit cf7742e0 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change

PiperOrigin-RevId: 363083122
parent 76f1113e
......@@ -475,6 +475,7 @@ class Resnet50CtlBenchmarkBase(CtlBenchmark):
self._set_df_common()
FLAGS.batch_size = 8192
FLAGS.dtype = 'bf16'
FLAGS.model_dir = self._get_model_dir('benchmark_8x16_tpu_bf16')
self._run_and_report_benchmark()
def fill_report_object(self, stats):
......
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