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

Add model card creation snippet to example scripts (#13730)

Browse Source 
* Update run_glue.py

* Update run_glue.py

* Add model creation snippet to other scripts

* Fix style
This commit is contained in:
Gunjan Chhablani
2021-09-24 19:21:46 +05:30
committed by GitHub
parent 66b01ce864
commit 38580455de
10 changed files with 113 additions and 91 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
examples/pytorch/text-classification/run_glue.py
View File

@@ -546,15 +546,17 @@ def main():
item = label_list[item]
writer.write(f"{index}\t{item}\n")
if training_args.push_to_hub:
kwargs = {"finetuned_from": model_args.model_name_or_path, "tasks": "text-classification"}
if data_args.task_name is not None:
kwargs["language"] = "en"
kwargs["dataset_tags"] = "glue"
kwargs["dataset_args"] = data_args.task_name
kwargs["dataset"] = f"GLUE {data_args.task_name.upper()}"
kwargs = {"finetuned_from": model_args.model_name_or_path, "tasks": "text-classification"}
if data_args.task_name is not None:
kwargs["language"] = "en"
kwargs["dataset_tags"] = "glue"
kwargs["dataset_args"] = data_args.task_name
kwargs["dataset"] = f"GLUE {data_args.task_name.upper()}"
if training_args.push_to_hub:
trainer.push_to_hub(**kwargs)
else:
trainer.create_model_card(**kwargs)
def _mp_fn(index):