config.json
1.2 KB · 60 lines · json Raw
1 {
2 "architectures": [
3 "AutoModel"
4 ],
5 "attention_probs_dropout_prob": 0.1,
6 "bos_token_id": null,
7 "do_sample": false,
8 "eos_token_ids": null,
9 "finetuning_task": null,
10 "hidden_act": "gelu",
11 "hidden_dropout_prob": 0.1,
12 "hidden_size": 768,
13 "id2label": {
14 "0": "O",
15 "1": "VENDOR",
16 "2": "DATE",
17 "3": "HCW",
18 "4": "HOSPITAL",
19 "5": "ID",
20 "6": "PATIENT",
21 "7": "PHONE"
22 },
23 "initializer_range": 0.02,
24 "intermediate_size": 3072,
25 "is_decoder": false,
26 "label2id": {
27 "O": 0,
28 "VENDOR": 1,
29 "DATE": 2,
30 "HCW": 3,
31 "HOSPITAL": 4,
32 "ID": 5,
33 "PATIENT": 6,
34 "PHONE": 7
35 },
36 "layer_norm_eps": 1e-12,
37 "length_penalty": 1.0,
38 "max_length": 20,
39 "max_position_embeddings": 512,
40 "model_type": "bert",
41 "num_attention_heads": 12,
42 "num_beams": 1,
43 "num_hidden_layers": 12,
44 "num_labels": 8,
45 "num_return_sequences": 1,
46 "output_attentions": false,
47 "output_hidden_states": false,
48 "output_past": true,
49 "pad_token_id": 0,
50 "pruned_heads": {},
51 "repetition_penalty": 1.0,
52 "temperature": 1.0,
53 "top_k": 50,
54 "top_p": 1.0,
55 "torchscript": false,
56 "type_vocab_size": 2,
57 "use_bfloat16": false,
58 "vocab_size": 30522
59 }
60