SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Update run_speech_recognition_seq2seq.py (#14967)

Browse Source
This commit is contained in:
flozi00
2022-01-06 17:26:45 +01:00
committed by GitHub
parent f71fb5c36e
commit b67f345d00
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/pytorch/speech-recognition/run_speech_recognition_seq2seq.py
View File

@@ -359,7 +359,7 @@ def main():
raw_datasets["train"] = raw_datasets["train"].select(range(data_args.max_train_samples)) raw_datasets["train"] = raw_datasets["train"].select(range(data_args.max_train_samples))
if data_args.max_eval_samples is not None: 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): def prepare_dataset(batch):
# process audio # process audio