From b67f345d00ef69fe2ba9f1d57ce6162cef785629 Mon Sep 17 00:00:00 2001 From: flozi00 Date: Thu, 6 Jan 2022 17:26:45 +0100 Subject: [PATCH] Update run_speech_recognition_seq2seq.py (#14967) --- .../speech-recognition/run_speech_recognition_seq2seq.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/pytorch/speech-recognition/run_speech_recognition_seq2seq.py b/examples/pytorch/speech-recognition/run_speech_recognition_seq2seq.py index 89efdca6a6..7aa410c6be 100755 --- a/examples/pytorch/speech-recognition/run_speech_recognition_seq2seq.py +++ b/examples/pytorch/speech-recognition/run_speech_recognition_seq2seq.py @@ -359,7 +359,7 @@ def main(): raw_datasets["train"] = raw_datasets["train"].select(range(data_args.max_train_samples)) if data_args.max_eval_samples is not None: - raw_datasets["eval"] = raw_datasets["eval"].select(range(data_args.max_train_samples)) + raw_datasets["eval"] = raw_datasets["eval"].select(range(data_args.max_eval_samples)) def prepare_dataset(batch): # process audio