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

Add id2label and label2id to config in run_xnil
2023-04-04 15:28:57 +02:00
parent fa2bdffc5d
commit 98268b2e76
1 changed files with 2 additions and 0 deletions
--- 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,