AliArshad commited on
Commit
d112e06
1 Parent(s): 9ea96f4

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +31 -8
config.json CHANGED
@@ -2,13 +2,36 @@
2
  "architectures": [
3
  "XLNetForSequenceClassification"
4
  ],
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5
  "model_type": "xlnet",
6
- "xlnet_version": "xlnet-base-cased",
7
- "num_labels": 2,
8
- "vocab_size": 32000,
9
- "d_model": 768,
10
- "num_heads": 12,
11
- "num_layers": 12,
12
- "max_position_embeddings": 512,
13
- "dropout": 0.1
 
 
 
 
 
 
 
 
 
14
  }
 
2
  "architectures": [
3
  "XLNetForSequenceClassification"
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": 4096,
11
+ "d_model": 1024,
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": 16,
21
+ "n_layer": 24,
22
+ "pad_token_id": 5,
23
+ "reuse_len": null,
24
+ "same_length": false,
25
+ "start_n_top": 5,
26
+ "summary_activation": null,
27
+ "summary_last_dropout": 0.1,
28
+ "summary_type": "last",
29
+ "summary_use_proj": true,
30
+ "task_specific_params": {
31
+ "text-generation": {
32
+ "do_sample": true,
33
+ "max_length": 50
34
+ }
35
+ },
36
+ "vocab_size": 32000
37
  }