onnx/config.json
| 1 | { |
| 2 | "_name_or_path": "dslim/bert-base-NER", |
| 3 | "_num_labels": 9, |
| 4 | "architectures": [ |
| 5 | "BertForTokenClassification" |
| 6 | ], |
| 7 | "attention_probs_dropout_prob": 0.1, |
| 8 | "classifier_dropout": null, |
| 9 | "hidden_act": "gelu", |
| 10 | "hidden_dropout_prob": 0.1, |
| 11 | "hidden_size": 768, |
| 12 | "id2label": { |
| 13 | "0": "O", |
| 14 | "1": "B-MISC", |
| 15 | "2": "I-MISC", |
| 16 | "3": "B-PER", |
| 17 | "4": "I-PER", |
| 18 | "5": "B-ORG", |
| 19 | "6": "I-ORG", |
| 20 | "7": "B-LOC", |
| 21 | "8": "I-LOC" |
| 22 | }, |
| 23 | "initializer_range": 0.02, |
| 24 | "intermediate_size": 3072, |
| 25 | "label2id": { |
| 26 | "B-LOC": 7, |
| 27 | "B-MISC": 1, |
| 28 | "B-ORG": 5, |
| 29 | "B-PER": 3, |
| 30 | "I-LOC": 8, |
| 31 | "I-MISC": 2, |
| 32 | "I-ORG": 6, |
| 33 | "I-PER": 4, |
| 34 | "O": 0 |
| 35 | }, |
| 36 | "layer_norm_eps": 1e-12, |
| 37 | "max_position_embeddings": 512, |
| 38 | "model_type": "bert", |
| 39 | "num_attention_heads": 12, |
| 40 | "num_hidden_layers": 12, |
| 41 | "output_past": true, |
| 42 | "pad_token_id": 0, |
| 43 | "position_embedding_type": "absolute", |
| 44 | "transformers_version": "4.34.0", |
| 45 | "type_vocab_size": 2, |
| 46 | "use_cache": true, |
| 47 | "vocab_size": 28996 |
| 48 | } |
| 49 | |