diff --git a/config.json b/config.json index ae22cc1..db8b75f 100644 --- a/config.json +++ b/config.json @@ -11,7 +11,7 @@ "hidden_dropout_prob": 0.1, "hidden_size": 768, "id2label": { - "0": "0", + "0": "O", "1": "VENDOR", "2": "DATE", "3": "HCW", @@ -24,7 +24,7 @@ "intermediate_size": 3072, "is_decoder": false, "label2id": { - "0": 0, + "O": 0, "VENDOR": 1, "DATE": 2, "HCW": 3,