Internal change

PiperOrigin-RevId: 410337787

official/benchmark/resnet_ctl_imagenet_benchmark.py

@@ -409,7 +409,7 @@ class Resnet50CtlBenchmarkBase(CtlBenchmark):
     FLAGS.single_l2_loss_op = True
     FLAGS.use_tf_function = True
     FLAGS.enable_checkpoint_and_export = False
-    FLAGS.data_dir = 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord'
+    FLAGS.data_dir = '/readahead/400M/placer/prod/home/distbelief/imagenet-tensorflow/imagenet-2012-tfrecord'
 
   def benchmark_2x2_tpu_bf16(self):
     self._setup()
@@ -436,6 +436,11 @@ class Resnet50CtlBenchmarkBase(CtlBenchmark):
     FLAGS.train_epochs = 4
     FLAGS.dtype = 'bf16'
     FLAGS.model_dir = self._get_model_dir('benchmark_4x4_tpu_bf16')
+    # Use dataset local to running cell (me). Also use ssd because this has
+    # given the max performance for ML Perf runs.
+    # TODO(emizan) See performance when placer has data on me cell and
+    # update copybara again
+    FLAGS.data_dir = 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord'
     self._run_and_report_benchmark()
 
   @owner_utils.Owner('tf-graph-compiler')