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_size": 768,
|
||||
"id2label": {
|
||||
"0": "LABEL_0",
|
||||
"1": "LABEL_1",
|
||||
"2": "LABEL_2"
|
||||
"0": "contradiction",
|
||||
"1": "neutral",
|
||||
"2": "entailment"
|
||||
},
|
||||
"initializer_range": 0.02,
|
||||
"intermediate_size": 3072,
|
||||
"label2id": {
|
||||
"LABEL_0": 0,
|
||||
"LABEL_1": 1,
|
||||
"LABEL_2": 2
|
||||
"contradiction": 0,
|
||||
"neutral": 1,
|
||||
"entailment": 2
|
||||
},
|
||||
"layer_norm_eps": 1e-12,
|
||||
"max_position_embeddings": 512,
|
||||
|
|
Loading…
Reference in New Issue