system
Update config.json 6ace25c
1 {
2 "architectures": [
3 "BertForPreTraining"
4 ],
5 "attention_probs_dropout_prob": 0.1,
6 "bos_token_id": 0,
7 "do_sample": false,
8 "eos_token_ids": 0,
9 "finetuning_task": null,
10 "hidden_act": "gelu",
11 "hidden_dropout_prob": 0.1,
12 "hidden_size": 768,
13 "id2label": {
14 "0": "LABEL_0",
15 "1": "LABEL_1"
16 },
17 "initializer_range": 0.02,
18 "intermediate_size": 3072,
19 "is_decoder": false,
20 "label2id": {
21 "LABEL_0": 0,
22 "LABEL_1": 1
23 },
24 "layer_norm_eps": 1e-12,
25 "length_penalty": 1.0,
26 "max_length": 20,
27 "max_position_embeddings": 512,
28 "model_type": "bert",
29 "num_attention_heads": 12,
30 "num_beams": 1,
31 "num_hidden_layers": 12,
32 "num_labels": 2,
33 "num_return_sequences": 1,
34 "output_attentions": false,
35 "output_hidden_states": false,
36 "output_past": true,
37 "pad_token_id": 0,
38 "pruned_heads": {},
39 "repetition_penalty": 1.0,
40 "temperature": 1.0,
41 "top_k": 50,
42 "top_p": 1.0,
43 "torchscript": false,
44 "type_vocab_size": 2,
45 "use_bfloat16": false,
46 "vocab_size": 30522
47 }
48