riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 18:51:14 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

[ASR] Update 'tasks' for model card (#19986)

Browse Source
This commit is contained in:
Sanchit Gandhi 2022-10-31 16:50:17 +00:00 committed by GitHub
parent 9406c7bc82
commit f38a145418
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -749,7 +749,7 @@ def main():
config_name = data_args.dataset_config_name if data_args.dataset_config_name is not None else "na" 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": "automatic-speech-recognition",
"tags": ["automatic-speech-recognition", data_args.dataset_name], "tags": ["automatic-speech-recognition", data_args.dataset_name],
"dataset_args": ( "dataset_args": (
f"Config: {config_name}, Training split: {data_args.train_split_name}, Eval split:" f"Config: {config_name}, Training split: {data_args.train_split_name}, Eval split:"

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

@ -503,7 +503,7 @@ def main():
trainer.save_metrics("eval", metrics) trainer.save_metrics("eval", metrics)
# 14. Write Training Stats # 14. Write Training Stats
kwargs = {"finetuned_from": model_args.model_name_or_path, "tasks": "speech recognition"} kwargs = {"finetuned_from": model_args.model_name_or_path, "tasks": "automatic-speech-recognition"}
if data_args.dataset_name is not None: if data_args.dataset_name is not None:
kwargs["dataset_tags"] = data_args.dataset_name kwargs["dataset_tags"] = data_args.dataset_name
if data_args.dataset_config_name is not None: if data_args.dataset_config_name is not None:

2
examples/research_projects/robust-speech-event/run_speech_recognition_ctc_bnb.py
View File

@ -758,7 +758,7 @@ def main():
config_name = data_args.dataset_config_name if data_args.dataset_config_name is not None else "na" 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": "automatic-speech-recognition",
"tags": ["automatic-speech-recognition", data_args.dataset_name], "tags": ["automatic-speech-recognition", data_args.dataset_name],
"dataset_args": ( "dataset_args": (
f"Config: {config_name}, Training split: {data_args.train_split_name}, Eval split:" f"Config: {config_name}, Training split: {data_args.train_split_name}, Eval split:"

2
examples/research_projects/robust-speech-event/run_speech_recognition_ctc_streaming.py
View File

@ -657,7 +657,7 @@ def main():
config_name = data_args.dataset_config_name if data_args.dataset_config_name is not None else "na" 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": "automatic-speech-recognition",
"tags": ["automatic-speech-recognition", data_args.dataset_name], "tags": ["automatic-speech-recognition", data_args.dataset_name],
"dataset_args": ( "dataset_args": (
f"Config: {config_name}, Training split: {data_args.train_split_name}, Eval split:" f"Config: {config_name}, Training split: {data_args.train_split_name}, Eval split:"