specifying librispeech dataset version in input pipeline since a new …

…version was added in tfds repo PiperOrigin-RevId: 493768612
google · Dec 8, 2022 · db578f5 · db578f5
1 parent b39d231
commit db578f5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/init2winit/dataset_lib/librispeech_input_pipeline.py b/init2winit/dataset_lib/librispeech_input_pipeline.py
@@ -274,7 +274,7 @@ def preprocess_data(
 def get_librispeech_datasets(hps, per_host_batch_size, per_host_eval_batch_size,
                              shuffle_rng):
   """Helper method to get train, eval and test sets for librispeech data."""
-  train_ds_builder = tfds.builder('librispeech')
+  train_ds_builder = tfds.builder('librispeech/2.1.1')
 
   train_data = get_raw_dataset(train_ds_builder, hps.train_split,
                                shuffle_rng[0])