fix/specify label ids
This commit is contained in:
parent
435eb54671
commit
e023a6ab02
12
config.json
12
config.json
|
@ -9,16 +9,16 @@
|
||||||
"hidden_dropout_prob": 0.1,
|
"hidden_dropout_prob": 0.1,
|
||||||
"hidden_size": 768,
|
"hidden_size": 768,
|
||||||
"id2label": {
|
"id2label": {
|
||||||
"0": "LABEL_0",
|
"0": "contradiction",
|
||||||
"1": "LABEL_1",
|
"1": "neutral",
|
||||||
"2": "LABEL_2"
|
"2": "entailment"
|
||||||
},
|
},
|
||||||
"initializer_range": 0.02,
|
"initializer_range": 0.02,
|
||||||
"intermediate_size": 3072,
|
"intermediate_size": 3072,
|
||||||
"label2id": {
|
"label2id": {
|
||||||
"LABEL_0": 0,
|
"contradiction": 0,
|
||||||
"LABEL_1": 1,
|
"neutral": 1,
|
||||||
"LABEL_2": 2
|
"entailment": 2
|
||||||
},
|
},
|
||||||
"layer_norm_eps": 1e-12,
|
"layer_norm_eps": 1e-12,
|
||||||
"max_position_embeddings": 512,
|
"max_position_embeddings": 512,
|
||||||
|
|
Loading…
Reference in New Issue