system HF staff commited on
Commit
f9ff0c8
1 Parent(s): dad80f7

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +4 -4
config.json CHANGED
@@ -1,18 +1,18 @@
1
  {
2
  "architectures": [
3
- "ElectraForMaskedLM"
4
  ],
5
  "attention_probs_dropout_prob": 0.1,
6
  "embedding_size": 768,
7
  "hidden_act": "gelu",
8
  "hidden_dropout_prob": 0.1,
9
- "hidden_size": 256,
10
  "initializer_range": 0.02,
11
- "intermediate_size": 1024,
12
  "layer_norm_eps": 1e-12,
13
  "max_position_embeddings": 512,
14
  "model_type": "electra",
15
- "num_attention_heads": 4,
16
  "num_hidden_layers": 12,
17
  "pad_token_id": 0,
18
  "type_vocab_size": 2,
1
  {
2
  "architectures": [
3
+ "ElectraForPreTraining"
4
  ],
5
  "attention_probs_dropout_prob": 0.1,
6
  "embedding_size": 768,
7
  "hidden_act": "gelu",
8
  "hidden_dropout_prob": 0.1,
9
+ "hidden_size": 768,
10
  "initializer_range": 0.02,
11
+ "intermediate_size": 3072,
12
  "layer_norm_eps": 1e-12,
13
  "max_position_embeddings": 512,
14
  "model_type": "electra",
15
+ "num_attention_heads": 12,
16
  "num_hidden_layers": 12,
17
  "pad_token_id": 0,
18
  "type_vocab_size": 2,