Skip to content

GitLab

Commit 8b510f95 authored by Allen Wang's avatar Allen Wang Committed by A. Unique TensorFlower
Browse files

Internal changes. 

PiperOrigin-RevId: 321382083
parent 900b1e07
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/benchmark/resnet_ctl_imagenet_benchmark.py
View file @ 8b510f95
...@@ -387,7 +387,7 @@ class Resnet50CtlBenchmarkBase(CtlBenchmark): ...@@ -387,7 +387,7 @@ class Resnet50CtlBenchmarkBase(CtlBenchmark):
FLAGS.single_l2_loss_op = True FLAGS.single_l2_loss_op = True
FLAGS.use_tf_function = True FLAGS.use_tf_function = True
FLAGS.enable_checkpoint_and_export = False FLAGS.enable_checkpoint_and_export = False
FLAGS.data_dir = gs://mlcompass-data/imagenet/imagenet-2012-tfrecord FLAGS.data_dir = 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord'
def benchmark_2x2_tpu_bf16(self): def benchmark_2x2_tpu_bf16(self):
self._setup() self._setup()
......
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