diff --git a/config.json b/config.json index 0f1de0a..1830e8f 100644 --- a/config.json +++ b/config.json @@ -12,19 +12,19 @@ "hidden_size": 768, "id2label": { "0": "O", - "1": "LOC", - "2": "PER", - "3": "MISC", - "4": "ORG" + "1": "I-LOC", + "2": "I-PER", + "3": "I-MISC", + "4": "I-ORG" }, "initializer_range": 0.02, "intermediate_size": 3072, "label2id": { - "LOC": 1, - "MISC": 3, + "I-LOC": 1, + "I-MISC": 3, "O": 0, - "ORG": 4, - "PER": 2 + "I-ORG": 4, + "I-PER": 2 }, "layer_norm_eps": 1e-05, "max_position_embeddings": 514,