Add id2label and label2id to model's config in run_xnil (#22558)

Add id2label and label2id to config in run_xnil

Add id2label and label2id to model's config in run_xnil (#22558)
Add id2label and label2id to config in run_xnil
98268b2e · Maziyar Panahi · GitHub · fa2bdffc · 98268b2e
Unverified Commit 98268b2e authored Apr 04, 2023 by Maziyar Panahi Committed by GitHub Apr 04, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

examples/pytorch/text-classification/run_xnli.py examples/pytorch/text-classification/run_xnli.py +2 -0

No files found.
--- a/examples/pytorch/text-classification/run_xnli.py
+++ b/examples/pytorch/text-classification/run_xnli.py
@@ -273,6 +273,8 @@ def main():
    config = AutoConfig.from_pretrained(
        model_args.config_name if model_args.config_name else model_args.model_name_or_path,
        num_labels=num_labels,
+        id2label={str(i): label for i, label in enumerate(label_list)},
+        label2id={label: i for i, label in enumerate(label_list)},
        finetuning_task="xnli",
        cache_dir=model_args.cache_dir,
        revision=model_args.model_revision,