Internal

PiperOrigin-RevId: 689058011
google · Oct 23, 2024 · 42d99a8 · 42d99a8
1 parent 0d2b495
commit 42d99a8
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',
+        tfds_dataset_name='imagenet_v2/matched-frequency:3.0.0',
         global_step=global_step,
     )
     test_ds = tfds.as_numpy(test_ds)