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

up (#14093)

Browse Source
This commit is contained in:
Patrick von Platen
2021-10-21 10:30:02 +02:00
committed by GitHub
parent 49155d2431
commit e9d2a639f4
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
examples/pytorch/speech-recognition/run_speech_recognition_ctc.py
View File

@@ -592,15 +592,16 @@ def main():
trainer.save_metrics("eval", metrics) trainer.save_metrics("eval", metrics)
# Write model card and (optionally) push to hub # 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 = { kwargs = {
"finetuned_from": model_args.model_name_or_path, "finetuned_from": model_args.model_name_or_path,
"tasks": "speech-recognition", "tasks": "speech-recognition",
"tags": ["automatic-speech-recognition", data_args.dataset_name], "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_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()} - {data_args.dataset_config_name.upper()}", "dataset": f"{data_args.dataset_name.upper()} - {config_name.upper()}",
} }
if "common_voice" in data_args.dataset_name: 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: if training_args.push_to_hub:
trainer.push_to_hub(**kwargs) trainer.push_to_hub(**kwargs)