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

🐛 fix small model card bugs (#13310)

Browse Source 
* 🐛 fix small model card bugs

* 💄 style
This commit is contained in:
Nathan Raw
2021-08-30 08:45:57 -06:00
committed by GitHub
parent 139e830158
commit ce6add8ecc
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/transformers/modelcard.py
View File

@@ -426,12 +426,15 @@ class TrainingSummary:
result["dataset"]["args"] = dataset_arg_mapping[ds_tag]
if len(metric_mapping) > 0:
result["metrics"] = []
for metric_tag, metric_name in metric_mapping.items():
result["metric"] = {
result["metrics"].append(
{
"name": metric_name,
"type": metric_tag,
"value": self.eval_results[metric_name],
}
)
model_index["results"].append(result)
@@ -446,7 +449,7 @@ class TrainingSummary:
metadata = _insert_values_as_list(metadata, "tags", self.tags)
metadata = _insert_values_as_list(metadata, "datasets", self.dataset_tags)
metadata = _insert_values_as_list(metadata, "metrics", list(metric_mapping.keys()))
metadata["model_index"] = self.create_model_index(metric_mapping)
metadata["model-index"] = self.create_model_index(metric_mapping)
return metadata