From 42d99a858843f37f6e278382f0c8e7e642720e14 Mon Sep 17 00:00:00 2001 From: Sourabh Medapati Date: Wed, 23 Oct 2024 12:15:47 -0700 Subject: [PATCH] Internal PiperOrigin-RevId: 689058011 --- init2winit/dataset_lib/imagenet_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/init2winit/dataset_lib/imagenet_dataset.py b/init2winit/dataset_lib/imagenet_dataset.py index 7ea7a650..4b826653 100644 --- 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)