Internal change

PiperOrigin-RevId: 480221272

Author: arashwan

official/projects/pruning/configs/experiments/image_classification/imagenet_mobilenetv2_pruning_gpu.yaml

@@ -18,12 +18,12 @@ task:
     one_hot: true
     label_smoothing: 0.1
   train_data:
-    input_path: 'imagenet-2012-tfrecord/train*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/train*'
     is_training: true
     global_batch_size: 1024
     dtype: 'float32'
   validation_data:
-    input_path: 'imagenet-2012-tfrecord/valid*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/valid*'
     is_training: false
     global_batch_size: 1024
     dtype: 'float32'

official/projects/pruning/configs/experiments/image_classification/imagenet_resnet50_pruning_gpu.yaml

@@ -15,12 +15,12 @@ task:
     one_hot: true
     label_smoothing: 0.1
   train_data:
-    input_path: 'imagenet-2012-tfrecord/train*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/train*'
     is_training: true
     global_batch_size: 1024
     dtype: 'float32'
   validation_data:
-    input_path: 'imagenet-2012-tfrecord/valid*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/valid*'
     is_training: false
     global_batch_size: 1024
     dtype: 'float32'

official/vision/configs/experiments/image_classification/imagenet_mobilenetv1_tpu.yaml

@@ -17,12 +17,12 @@ task:
     one_hot: true
     label_smoothing: 0.1
   train_data:
-    input_path: 'imagenet-2012-tfrecord/train*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/train*'
     is_training: true
     global_batch_size: 4096
     dtype: 'bfloat16'
   validation_data:
-    input_path: 'imagenet-2012-tfrecord/valid*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/valid*'
     is_training: false
     global_batch_size: 4096
     dtype: 'bfloat16'

official/vision/configs/experiments/image_classification/imagenet_mobilenetv2_gpu.yaml

@@ -18,12 +18,12 @@ task:
     one_hot: true
     label_smoothing: 0.1
   train_data:
-    input_path: 'imagenet-2012-tfrecord/train*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/train*'
     is_training: true
     global_batch_size: 1024  # 128 * 8
     dtype: 'float16'
   validation_data:
-    input_path: 'imagenet-2012-tfrecord/valid*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/valid*'
     is_training: false
     global_batch_size: 1024  # 128 * 8
     dtype: 'float16'

official/vision/configs/experiments/image_classification/imagenet_mobilenetv2_tpu.yaml

@@ -17,12 +17,12 @@ task:
     one_hot: true
     label_smoothing: 0.1
   train_data:
-    input_path: 'imagenet-2012-tfrecord/train*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/train*'
     is_training: true
     global_batch_size: 4096
     dtype: 'bfloat16'
   validation_data:
-    input_path: 'imagenet-2012-tfrecord/valid*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/valid*'
     is_training: false
     global_batch_size: 4096
     dtype: 'bfloat16'

official/vision/configs/experiments/image_classification/imagenet_mobilenetv3.5_avg_tpu.yaml

@@ -23,12 +23,12 @@ task:
     one_hot: true
     label_smoothing: 0.1
   train_data:
-    input_path: 'imagenet-2012-tfrecord/train*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/train*'
     is_training: true
     global_batch_size: 4096
     dtype: 'bfloat16'
   validation_data:
-    input_path: 'imagenet-2012-tfrecord/valid*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/valid*'
     is_training: false
     global_batch_size: 4096
     dtype: 'bfloat16'

official/vision/configs/experiments/image_classification/imagenet_mobilenetv3.5_avgseg_tpu.yaml

@@ -23,12 +23,12 @@ task:
     one_hot: true
     label_smoothing: 0.1
   train_data:
-    input_path: 'imagenet-2012-tfrecord/train*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/train*'
     is_training: true
     global_batch_size: 4096
     dtype: 'bfloat16'
   validation_data:
-    input_path: 'imagenet-2012-tfrecord/valid*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/valid*'
     is_training: false
     global_batch_size: 4096
     dtype: 'bfloat16'

official/vision/configs/experiments/image_classification/imagenet_mobilenetv3large_tpu.yaml

@@ -21,7 +21,7 @@ task:
     one_hot: true
     label_smoothing: 0.1
   train_data:
-    input_path: 'imagenet-2012-tfrecord/train*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/train*'
     is_training: true
     global_batch_size: 4096
     dtype: 'bfloat16'
@@ -32,7 +32,7 @@ task:
         translate_const: 250
       type: 'autoaug'
   validation_data:
-    input_path: 'imagenet-2012-tfrecord/valid*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/valid*'
     is_training: false
     global_batch_size: 4096
     dtype: 'bfloat16'

official/vision/configs/experiments/image_classification/imagenet_mobilenetv3small_tpu.yaml

@@ -22,12 +22,12 @@ task:
     one_hot: true
     label_smoothing: 0.1
   train_data:
-    input_path: 'imagenet-2012-tfrecord/train*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/train*'
     is_training: true
     global_batch_size: 4096
     dtype: 'bfloat16'
   validation_data:
-    input_path: 'imagenet-2012-tfrecord/valid*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/valid*'
     is_training: false
     global_batch_size: 4096
     dtype: 'bfloat16'

official/vision/configs/experiments/image_classification/imagenet_resnet101_deeplab_tpu.yaml

@@ -23,13 +23,13 @@ task:
     one_hot: true
     label_smoothing: 0.1
   train_data:
-    input_path: 'imagenet-2012-tfrecord/train*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/train*'
     is_training: true
     global_batch_size: 4096
     dtype: 'bfloat16'
     aug_policy: 'randaug'
   validation_data:
-    input_path: 'imagenet-2012-tfrecord/valid*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/valid*'
     is_training: false
     global_batch_size: 4096
     dtype: 'bfloat16'

official/vision/configs/experiments/image_classification/imagenet_resnet101_tpu.yaml

@@ -17,12 +17,12 @@ task:
     one_hot: true
     label_smoothing: 0.1
   train_data:
-    input_path: 'imagenet-2012-tfrecord/train*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/train*'
     is_training: true
     global_batch_size: 4096
     dtype: 'bfloat16'
   validation_data:
-    input_path: 'imagenet-2012-tfrecord/valid*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/valid*'
     is_training: false
     global_batch_size: 4096
     dtype: 'bfloat16'

official/vision/configs/experiments/image_classification/imagenet_resnet152_tpu.yaml

@@ -17,12 +17,12 @@ task:
     one_hot: true
     label_smoothing: 0.1
   train_data:
-    input_path: 'imagenet-2012-tfrecord/train*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/train*'
     is_training: true
     global_batch_size: 4096
     dtype: 'bfloat16'
   validation_data:
-    input_path: 'imagenet-2012-tfrecord/valid*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/valid*'
     is_training: false
     global_batch_size: 4096
     dtype: 'bfloat16'

official/vision/configs/experiments/image_classification/imagenet_resnet50_deeplab_tpu.yaml

@@ -17,12 +17,12 @@ task:
     one_hot: true
     label_smoothing: 0.1
   train_data:
-    input_path: 'imagenet-2012-tfrecord/train*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/train*'
     is_training: true
     global_batch_size: 4096
     dtype: 'bfloat16'
   validation_data:
-    input_path: 'imagenet-2012-tfrecord/valid*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/valid*'
     is_training: false
     global_batch_size: 4096
     dtype: 'bfloat16'

official/vision/configs/experiments/image_classification/imagenet_resnet50_gpu.yaml

@@ -15,15 +15,15 @@ task:
     one_hot: true
     label_smoothing: 0.1
   train_data:
-    input_path: 'imagenet-2012-tfrecord/train*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/train*'
     is_training: true
     global_batch_size: 2048
     dtype: 'float16'
     # Autotuning the prefetch buffer size causes OOMs, so set it to a reasonable
     # static value: 32. See b/218880025.
     prefetch_buffer_size: 32
   validation_data:
-    input_path: 'imagenet-2012-tfrecord/valid*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/valid*'
     is_training: false
     global_batch_size: 2048
     dtype: 'float16'

official/vision/configs/experiments/image_classification/imagenet_resnet50_tpu.yaml

@@ -14,12 +14,12 @@ task:
     one_hot: true
     label_smoothing: 0.1
   train_data:
-    input_path: 'imagenet-2012-tfrecord/train*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/train*'
     is_training: true
     global_batch_size: 4096
     dtype: 'bfloat16'
   validation_data:
-    input_path: 'imagenet-2012-tfrecord/valid*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/valid*'
     is_training: false
     global_batch_size: 4096
     dtype: 'bfloat16'

official/vision/configs/experiments/image_classification/imagenet_resnetrs101_i160.yaml

@@ -25,7 +25,7 @@ task:
     one_hot: true
     label_smoothing: 0.1
   train_data:
-    input_path: 'imagenet-2012-tfrecord/train*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/train*'
     is_training: true
     global_batch_size: 4096
     dtype: 'bfloat16'
@@ -34,7 +34,7 @@ task:
       randaug:
         magnitude: 15
   validation_data:
-    input_path: 'imagenet-2012-tfrecord/valid*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/valid*'
     is_training: false
     global_batch_size: 4096
     dtype: 'bfloat16'

official/vision/configs/experiments/image_classification/imagenet_resnetrs101_i192.yaml

@@ -25,7 +25,7 @@ task:
     one_hot: true
     label_smoothing: 0.1
   train_data:
-    input_path: 'imagenet-2012-tfrecord/train*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/train*'
     is_training: true
     global_batch_size: 4096
     dtype: 'bfloat16'
@@ -34,7 +34,7 @@ task:
       randaug:
         magnitude: 15
   validation_data:
-    input_path: 'imagenet-2012-tfrecord/valid*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/valid*'
     is_training: false
     global_batch_size: 4096
     dtype: 'bfloat16'

official/vision/configs/experiments/image_classification/imagenet_resnetrs152_i192.yaml

@@ -25,7 +25,7 @@ task:
     one_hot: true
     label_smoothing: 0.1
   train_data:
-    input_path: 'imagenet-2012-tfrecord/train*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/train*'
     is_training: true
     global_batch_size: 4096
     dtype: 'bfloat16'
@@ -34,7 +34,7 @@ task:
       randaug:
         magnitude: 15
   validation_data:
-    input_path: 'imagenet-2012-tfrecord/valid*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/valid*'
     is_training: false
     global_batch_size: 4096
     dtype: 'bfloat16'

official/vision/configs/experiments/image_classification/imagenet_resnetrs152_i224.yaml

@@ -25,7 +25,7 @@ task:
     one_hot: true
     label_smoothing: 0.1
   train_data:
-    input_path: 'imagenet-2012-tfrecord/train*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/train*'
     is_training: true
     global_batch_size: 4096
     dtype: 'bfloat16'
@@ -34,7 +34,7 @@ task:
       randaug:
         magnitude: 15
   validation_data:
-    input_path: 'imagenet-2012-tfrecord/valid*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/valid*'
     is_training: false
     global_batch_size: 4096
     dtype: 'bfloat16'

official/vision/configs/experiments/image_classification/imagenet_resnetrs152_i256.yaml

@@ -25,7 +25,7 @@ task:
     one_hot: true
     label_smoothing: 0.1
   train_data:
-    input_path: 'imagenet-2012-tfrecord/train*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/train*'
     is_training: true
     global_batch_size: 4096
     dtype: 'bfloat16'
@@ -34,7 +34,7 @@ task:
       randaug:
         magnitude: 15
   validation_data:
-    input_path: 'imagenet-2012-tfrecord/valid*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/valid*'
     is_training: false
     global_batch_size: 4096
     dtype: 'bfloat16'

official/vision/configs/experiments/image_classification/imagenet_resnetrs200_i256.yaml

@@ -25,7 +25,7 @@ task:
     one_hot: true
     label_smoothing: 0.1
   train_data:
-    input_path: 'imagenet-2012-tfrecord/train*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/train*'
     is_training: true
     global_batch_size: 4096
     dtype: 'bfloat16'
@@ -34,7 +34,7 @@ task:
       randaug:
         magnitude: 15
   validation_data:
-    input_path: 'imagenet-2012-tfrecord/valid*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/valid*'
     is_training: false
     global_batch_size: 4096
     dtype: 'bfloat16'

official/vision/configs/experiments/image_classification/imagenet_resnetrs270_i256.yaml

@@ -25,7 +25,7 @@ task:
     one_hot: true
     label_smoothing: 0.1
   train_data:
-    input_path: 'imagenet-2012-tfrecord/train*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/train*'
     is_training: true
     global_batch_size: 4096
     dtype: 'bfloat16'
@@ -34,7 +34,7 @@ task:
       randaug:
         magnitude: 15
   validation_data:
-    input_path: 'imagenet-2012-tfrecord/valid*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/valid*'
     is_training: false
     global_batch_size: 4096
     dtype: 'bfloat16'

official/vision/configs/experiments/image_classification/imagenet_resnetrs350_i256.yaml

@@ -25,7 +25,7 @@ task:
     one_hot: true
     label_smoothing: 0.1
   train_data:
-    input_path: 'imagenet-2012-tfrecord/train*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/train*'
     is_training: true
     global_batch_size: 4096
     dtype: 'bfloat16'
@@ -34,7 +34,7 @@ task:
       randaug:
         magnitude: 15
   validation_data:
-    input_path: 'imagenet-2012-tfrecord/valid*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/valid*'
     is_training: false
     global_batch_size: 4096
     dtype: 'bfloat16'

official/vision/configs/experiments/image_classification/imagenet_resnetrs350_i320.yaml

@@ -25,7 +25,7 @@ task:
     one_hot: true
     label_smoothing: 0.1
   train_data:
-    input_path: 'imagenet-2012-tfrecord/train*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/train*'
     is_training: true
     global_batch_size: 4096
     dtype: 'bfloat16'
@@ -34,7 +34,7 @@ task:
       randaug:
         magnitude: 15
   validation_data:
-    input_path: 'imagenet-2012-tfrecord/valid*'
+    input_path: 'gs://mlcompass-data/imagenet/imagenet-2012-tfrecord/valid*'
     is_training: false
     global_batch_size: 4096
     dtype: 'bfloat16'