jacoballessio commited on
Commit
643bb1f
·
verified ·
1 Parent(s): 472a907

Create config.json

Browse files
Files changed (1) hide show
  1. config.json +55 -0
config.json ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "AI Image Detect Distilled"
4
+ ],
5
+ "model_type": "AI Image Detect Distilled",
6
+ "num_labels": 2,
7
+ "id2label": {
8
+ "0": "fake",
9
+ "1": "real"
10
+ },
11
+ "label2id": {
12
+ "fake": 0,
13
+ "real": 1
14
+ },
15
+ "ensemble_models": [
16
+ {
17
+ "model_type": "vit",
18
+ "hidden_size": 768,
19
+ "num_hidden_layers": 12,
20
+ "num_attention_heads": 12,
21
+ "intermediate_size": 3072,
22
+ "hidden_act": "gelu",
23
+ "image_size": 224,
24
+ "patch_size": 16
25
+ },
26
+ {
27
+ "model_type": "resnet",
28
+ "depth": 50,
29
+ "num_channels": 3,
30
+ "embedding_size": 2048
31
+ },
32
+ {
33
+ "model_type": "efficientnet",
34
+ "width_coefficient": 1.4,
35
+ "depth_coefficient": 1.8,
36
+ "dropout_rate": 0.4
37
+ }
38
+ ],
39
+ "distilled_model": {
40
+ "model_type": "efficientnet",
41
+ "width_coefficient": 1.4,
42
+ "depth_coefficient": 1.8,
43
+ "dropout_rate": 0.4
44
+ },
45
+ "image_size": 224,
46
+ "num_channels": 3,
47
+ "dropout_rate": 0.2,
48
+ "classifier_dropout_prob": 0.1,
49
+ "initializer_range": 0.02,
50
+ "layer_norm_eps": 1e-12,
51
+ "is_encoder_decoder": false,
52
+ "use_cache": true,
53
+ "torch_dtype": "float32",
54
+ "transformers_version": "4.28.0"
55
+ }