fix: id2label and label2id

This commit is contained in:
David Chu 2021-02-13 18:32:43 +00:00 committed by huggingface-web
parent 25850c5418
commit b91e7a74c6
1 changed files with 5 additions and 5 deletions

View File

@ -10,15 +10,15 @@
"finetuning_task": "mnli",
"hidden_dim": 3072,
"id2label": {
"0": "CONTRADICTION",
"0": "ENTAILMENT",
"1": "NEUTRAL",
"2": "ENTAILMENT"
"2": "CONTRADICTION"
},
"initializer_range": 0.02,
"label2id": {
"CONTRADICTION": 0,
"ENTAILMENT": 2,
"NEUTRAL": 1
"ENTAILMENT": 0,
"NEUTRAL": 1,
"CONTRADICTION": 2,
},
"max_position_embeddings": 512,
"model_type": "distilbert",