diff --git a/config.json b/config.json index 0cd7199..109bddf 100644 --- a/config.json +++ b/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,