Add event tag to script
Browse files
run_speech_recognition_ctc.py
CHANGED
@@ -737,7 +737,7 @@ def main():
|
|
737 |
kwargs = {
|
738 |
"finetuned_from": model_args.model_name_or_path,
|
739 |
"tasks": "speech-recognition",
|
740 |
-
"tags": ["automatic-speech-recognition", data_args.dataset_name],
|
741 |
"dataset_args": f"Config: {config_name}, Training split: {data_args.train_split_name}, Eval split: {data_args.eval_split_name}",
|
742 |
"dataset": f"{data_args.dataset_name.upper()} - {config_name.upper()}",
|
743 |
}
|
|
|
737 |
kwargs = {
|
738 |
"finetuned_from": model_args.model_name_or_path,
|
739 |
"tasks": "speech-recognition",
|
740 |
+
"tags": ["automatic-speech-recognition", data_args.dataset_name, "robust-speech-event"],
|
741 |
"dataset_args": f"Config: {config_name}, Training split: {data_args.train_split_name}, Eval split: {data_args.eval_split_name}",
|
742 |
"dataset": f"{data_args.dataset_name.upper()} - {config_name.upper()}",
|
743 |
}
|
run_speech_recognition_ctc_bnb.py
CHANGED
@@ -761,7 +761,7 @@ def main():
|
|
761 |
kwargs = {
|
762 |
"finetuned_from": model_args.model_name_or_path,
|
763 |
"tasks": "speech-recognition",
|
764 |
-
"tags": ["automatic-speech-recognition", data_args.dataset_name],
|
765 |
"dataset_args": f"Config: {config_name}, Training split: {data_args.train_split_name}, Eval split: {data_args.eval_split_name}",
|
766 |
"dataset": f"{data_args.dataset_name.upper()} - {config_name.upper()}",
|
767 |
}
|
|
|
761 |
kwargs = {
|
762 |
"finetuned_from": model_args.model_name_or_path,
|
763 |
"tasks": "speech-recognition",
|
764 |
+
"tags": ["automatic-speech-recognition", data_args.dataset_name, "robust-speech-event"],
|
765 |
"dataset_args": f"Config: {config_name}, Training split: {data_args.train_split_name}, Eval split: {data_args.eval_split_name}",
|
766 |
"dataset": f"{data_args.dataset_name.upper()} - {config_name.upper()}",
|
767 |
}
|