Internal

PiperOrigin-RevId: 693540465
google · Nov 6, 2024 · a0d4718 · a0d4718
1 parent 42d99a8
commit a0d4718
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/init2winit/dataset_lib/imagenet_dataset.py b/init2winit/dataset_lib/imagenet_dataset.py
@@ -442,7 +442,7 @@ def get_imagenet(shuffle_rng, batch_size, eval_batch_size, hps, global_step=0):
         'test',
         hps=hps,
         image_size=image_size,
-        tfds_dataset_name='imagenet_v2/matched-frequency:3.0.0',
+        tfds_dataset_name='imagenet_v2/matched-frequency',
         global_step=global_step,
     )
     test_ds = tfds.as_numpy(test_ds)