fix: id2label and label2id
This commit is contained in:
parent
25850c5418
commit
b91e7a74c6
10
config.json
10
config.json
|
@ -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",
|
||||
|
|
Loading…
Reference in New Issue