system commited on
Commit
e1a20d7
1 Parent(s): 26a516d

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +7 -8
config.json CHANGED
@@ -4,23 +4,22 @@
4
  ],
5
  "attn_type": "bi",
6
  "bi_data": false,
 
7
  "clamp_len": -1,
8
  "d_head": 64,
9
  "d_inner": 3072,
10
  "d_model": 768,
11
  "dropout": 0.1,
12
  "end_n_top": 5,
 
13
  "ff_activation": "gelu",
14
- "finetuning_task": null,
15
  "initializer_range": 0.02,
16
  "layer_norm_eps": 1e-12,
17
  "mem_len": null,
 
18
  "n_head": 12,
19
  "n_layer": 12,
20
- "n_token": 32000,
21
- "num_labels": 2,
22
- "output_attentions": false,
23
- "output_hidden_states": false,
24
  "reuse_len": null,
25
  "same_length": false,
26
  "start_n_top": 5,
@@ -33,6 +32,6 @@
33
  "max_length": 210
34
  }
35
  },
36
- "torchscript": false,
37
- "untie_r": true
38
- }
4
  ],
5
  "attn_type": "bi",
6
  "bi_data": false,
7
+ "bos_token_id": 1,
8
  "clamp_len": -1,
9
  "d_head": 64,
10
  "d_inner": 3072,
11
  "d_model": 768,
12
  "dropout": 0.1,
13
  "end_n_top": 5,
14
+ "eos_token_id": 2,
15
  "ff_activation": "gelu",
 
16
  "initializer_range": 0.02,
17
  "layer_norm_eps": 1e-12,
18
  "mem_len": null,
19
+ "model_type": "xlnet",
20
  "n_head": 12,
21
  "n_layer": 12,
22
+ "pad_token_id": 5,
 
 
 
23
  "reuse_len": null,
24
  "same_length": false,
25
  "start_n_top": 5,
32
  "max_length": 210
33
  }
34
  },
35
+ "untie_r": true,
36
+ "vocab_size": 32000
37
+ }