kampkelly commited on
Commit
5556236
1 Parent(s): 87ffc3d

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +3 -3
config.json CHANGED
@@ -4,11 +4,11 @@
4
  ],
5
  "model_type": "EfficientNetB0",
6
  "num_labels": 7,
7
- "hidden_size": 1280,
8
  "num_attention_heads": 1,
9
- "num_hidden_layers": 0,
10
  "intermediate_size": 0,
11
- "hidden_act": "gelu",
12
  "hidden_dropout_prob": 0.2,
13
  "attention_probs_dropout_prob": 0.2,
14
  "initializer_range": 0.02,
 
4
  ],
5
  "model_type": "EfficientNetB0",
6
  "num_labels": 7,
7
+ "hidden_size": 128,
8
  "num_attention_heads": 1,
9
+ "num_hidden_layers": 2,
10
  "intermediate_size": 0,
11
+ "hidden_act": "relu",
12
  "hidden_dropout_prob": 0.2,
13
  "attention_probs_dropout_prob": 0.2,
14
  "initializer_range": 0.02,