diff --git a/config.json b/config.json index b89edbd..4d2f7cf 100644 --- a/config.json +++ b/config.json @@ -11,16 +11,16 @@ "hidden_dropout_prob": 0.1, "hidden_size": 768, "id2label": { - "0": "contradiction", + "0": "entailment", "1": "neutral", - "2": "entailment" + "2": "contradiction" }, "initializer_range": 0.02, "intermediate_size": 3072, "label2id": { - "contradiction": 0, + "entailment": 0, "neutral": 1, - "entailment": 2 + "contradiction": 2 }, "layer_norm_eps": 1e-05, "max_position_embeddings": 514,