jacoballessio
commited on
Update config.json
Browse files- config.json +2 -5
config.json
CHANGED
@@ -3,12 +3,8 @@
|
|
3 |
"architectures": [
|
4 |
"EfficientNetForImageClassification"
|
5 |
],
|
|
|
6 |
"num_labels": 2,
|
7 |
-
"attention_probs_dropout_prob": 0.0,
|
8 |
-
"encoder_stride": 16,
|
9 |
-
"hidden_act": "gelu",
|
10 |
-
"hidden_dropout_prob": 0.0,
|
11 |
-
"hidden_size": 768,
|
12 |
"id2label": {
|
13 |
"0": "fake",
|
14 |
"1": "real"
|
@@ -27,6 +23,7 @@
|
|
27 |
},
|
28 |
"classifier_dropout_prob": 0.2,
|
29 |
"hidden_act": "swish",
|
|
|
30 |
"torch_dtype": "float32",
|
31 |
"use_cache": true,
|
32 |
"output_hidden_states": false
|
|
|
3 |
"architectures": [
|
4 |
"EfficientNetForImageClassification"
|
5 |
],
|
6 |
+
"model_type": "efficientnet",
|
7 |
"num_labels": 2,
|
|
|
|
|
|
|
|
|
|
|
8 |
"id2label": {
|
9 |
"0": "fake",
|
10 |
"1": "real"
|
|
|
23 |
},
|
24 |
"classifier_dropout_prob": 0.2,
|
25 |
"hidden_act": "swish",
|
26 |
+
"transformers_version": "4.41.2",
|
27 |
"torch_dtype": "float32",
|
28 |
"use_cache": true,
|
29 |
"output_hidden_states": false
|