run_speech_recognition_seq2seq.py: add cache_dir param to dataset (#20540)

2022-12-07 13:23:16 -05:00
parent fc95386ea1
commit 0526a075c5
1 changed files with 2 additions and 0 deletions
--- a/examples/pytorch/speech-recognition/run_speech_recognition_seq2seq.py
+++ b/examples/pytorch/speech-recognition/run_speech_recognition_seq2seq.py
@@ -327,6 +327,7 @@ def main():
            data_args.dataset_name,
            data_args.dataset_config_name,
            split=data_args.train_split_name,
+            cache_dir=model_args.cache_dir,
            use_auth_token=True if model_args.use_auth_token else None,
        )

@@ -335,6 +336,7 @@ def main():
            data_args.dataset_name,
            data_args.dataset_config_name,
            split=data_args.eval_split_name,
+            cache_dir=model_args.cache_dir,
            use_auth_token=True if model_args.use_auth_token else None,
        )