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

* 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

View File

@@ -430,15 +430,19 @@ def main():
trainer.log_metrics("eval", metrics)
trainer.save_metrics("eval", metrics)
kwargs = dict(
finetuned_from=model_args.model_name_or_path,
tasks="multiple-choice",
dataset_tags="swag",
dataset_args="regular",
dataset="SWAG",
language="en",
)
if training_args.push_to_hub:
trainer.push_to_hub(
finetuned_from=model_args.model_name_or_path,
tasks="multiple-choice",
dataset_tags="swag",
dataset_args="regular",
dataset="SWAG",
language="en",
)
trainer.push_to_hub(**kwargs)
else:
trainer.create_model_card(**kwargs)
def _mp_fn(index):