suolyer commited on
Commit
7349e84
1 Parent(s): 2001c51

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +25 -0
config.json ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"architectures": [
2
+ "MegatronBertForSequenceClassification"
3
+ ],
4
+ "_num_labels": 3,
5
+ "id2label": {
6
+ "0": "CONTRADICTION",
7
+ "1": "NEUTRAL",
8
+ "2": "ENTAILMENT"
9
+ },
10
+ "vocab_size": 21248,
11
+ "hidden_size": 2048,
12
+ "num_hidden_layers": 24,
13
+ "num_attention_heads": 8,
14
+ "hidden_act": "gelu_new",
15
+ "intermediate_size": 8192,
16
+ "hidden_dropout_prob": 0.1,
17
+ "attention_probs_dropout_prob": 0.1,
18
+ "max_position_embeddings": 512,
19
+ "type_vocab_size": 2,
20
+ "initializer_range": 0.2,
21
+ "layer_norm_eps": 1e-12,
22
+ "gradient_checkpointing": false,
23
+ "position_embedding_type": "absolute",
24
+ "use_cache": false,
25
+ "model_type": "megatron-bert"}