From e9d2a639f4d13ca3959156c1b50f0ed6398b00eb Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Thu, 21 Oct 2021 10:30:02 +0200 Subject: [PATCH] up (#14093) --- .../speech-recognition/run_speech_recognition_ctc.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/examples/pytorch/speech-recognition/run_speech_recognition_ctc.py b/examples/pytorch/speech-recognition/run_speech_recognition_ctc.py index ca5841e32f..2b1a5810f7 100755 --- a/examples/pytorch/speech-recognition/run_speech_recognition_ctc.py +++ b/examples/pytorch/speech-recognition/run_speech_recognition_ctc.py @@ -592,15 +592,16 @@ def main(): trainer.save_metrics("eval", metrics) # Write model card and (optionally) push to hub + config_name = data_args.dataset_config_name if data_args.dataset_config_name is not None else "na" kwargs = { "finetuned_from": model_args.model_name_or_path, "tasks": "speech-recognition", "tags": ["automatic-speech-recognition", data_args.dataset_name], - "dataset_args": f"Config: {data_args.dataset_config_name}, Training split: {data_args.train_split_name}, Eval split: {data_args.eval_split_name}", - "dataset": f"{data_args.dataset_name.upper()} - {data_args.dataset_config_name.upper()}", + "dataset_args": f"Config: {config_name}, Training split: {data_args.train_split_name}, Eval split: {data_args.eval_split_name}", + "dataset": f"{data_args.dataset_name.upper()} - {config_name.upper()}", } if "common_voice" in data_args.dataset_name: - kwargs["language"] = data_args.dataset_config_name + kwargs["language"] = config_name if training_args.push_to_hub: trainer.push_to_hub(**kwargs)