diff --git a/config.json b/config.json new file mode 100644 index 0000000..8499135 --- /dev/null +++ b/config.json @@ -0,0 +1,48 @@ +{ + "attention_probs_dropout_prob": 0.1, + "directionality": "bidi", + "finetuning_task": null, + "hidden_act": "gelu", + "hidden_dropout_prob": 0.1, + "hidden_size": 1024, + "initializer_range": 0.02, + "intermediate_size": 4096, + "layer_norm_eps": 1e-12, + "max_position_embeddings": 512, + "num_attention_heads": 16, + "num_hidden_layers": 24, + "num_labels": 9, + "output_attentions": false, + "output_hidden_states": false, + "pooler_fc_size": 768, + "pooler_num_attention_heads": 12, + "pooler_num_fc_layers": 3, + "pooler_size_per_head": 128, + "pooler_type": "first_token_transform", + "pruned_heads": {}, + "torchscript": false, + "type_vocab_size": 2, + "vocab_size": 28996, + "label2id": { + "O": 0, + "B-MISC": 1, + "I-MISC": 2, + "B-PER": 3, + "I-PER": 4, + "B-ORG": 5, + "I-ORG": 6, + "B-LOC": 7, + "I-LOC": 8 + }, + "id2label": { + "0": "O", + "1": "B-MISC", + "2": "I-MISC", + "3": "B-PER", + "4": "I-PER", + "5": "B-ORG", + "6": "I-ORG", + "7": "B-LOC", + "8": "I-LOC" + } +}