From afae42a725e032de1c463eef91a78aaedd23d5f2 Mon Sep 17 00:00:00 2001 From: BDoyen Date: Thu, 25 Mar 2021 11:11:40 +0000 Subject: [PATCH] change config --- config.json | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/config.json b/config.json index b167c1b..26420da 100644 --- a/config.json +++ b/config.json @@ -10,16 +10,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,