Skip to content

GitLab

"...git@developer.sourcefind.cn:OpenDAS/mmdetection3d.git" did not exist on "9a30edfcbeca07ca48ddc5223290dd5646dee294"
Commit cf7742e0 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 363083122
parent 76f1113e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
official/benchmark/resnet_ctl_imagenet_benchmark.py
View file @ cf7742e0
...@@ -475,6 +475,7 @@ class Resnet50CtlBenchmarkBase(CtlBenchmark): ...@@ -475,6 +475,7 @@ class Resnet50CtlBenchmarkBase(CtlBenchmark):
self._set_df_common() self._set_df_common()
FLAGS.batch_size = 8192 FLAGS.batch_size = 8192
FLAGS.dtype = 'bf16' FLAGS.dtype = 'bf16'
FLAGS.model_dir = self._get_model_dir('benchmark_8x16_tpu_bf16')
self._run_and_report_benchmark() self._run_and_report_benchmark()
def fill_report_object(self, stats): 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