Hugging Face's logo

pabloOmega
/
egg_classifier

Image Classification
Transformers
Safetensors
efficientnet
Model card Files
xet
Community
pabloOmega commited on
Commit
664d9b3
·
verified ·
1 Parent(s): 21e1307

Upload EfficientNetForImageClassification

Browse files
Files changed (2) hide show
  1. config.json +82 -17
  2. model.safetensors +2 -2
config.json CHANGED
@@ -1,13 +1,26 @@
1
  {
2
  "architectures": [
3
- "ViTForImageClassification"
4
  ],
5
- "attention_probs_dropout_prob": 0.0,
 
 
 
 
 
 
6
  "dtype": "float32",
7
- "encoder_stride": 16,
8
- "hidden_act": "gelu",
9
- "hidden_dropout_prob": 0.0,
10
- "hidden_size": 768,
 
 
 
 
 
 
 
11
  "id2label": {
12
  "0": "class_12_25",
13
  "1": "class_13_25",
@@ -15,9 +28,26 @@
15
  "3": "class_15_25",
16
  "4": "class_16_25"
17
  },
18
- "image_size": 384,
 
 
 
 
 
 
 
 
 
19
  "initializer_range": 0.02,
20
- "intermediate_size": 3072,
 
 
 
 
 
 
 
 
21
  "label2id": {
22
  "class_12_25": 0,
23
  "class_13_25": 1,
@@ -25,15 +55,50 @@
25
  "class_15_25": 3,
26
  "class_16_25": 4
27
  },
28
- "layer_norm_eps": 1e-12,
29
- "model_type": "vit",
30
- "num_attention_heads": 12,
 
 
 
 
 
 
 
31
  "num_channels": 3,
32
- "num_hidden_layers": 12,
33
- "patch_size": 16,
34
- "pooler_act": "tanh",
35
- "pooler_output_size": 768,
 
 
 
 
 
 
 
 
36
  "problem_type": "single_label_classification",
37
- "qkv_bias": true,
38
- "transformers_version": "4.57.1"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
39
  }
 
1
  {
2
  "architectures": [
3
+ "EfficientNetForImageClassification"
4
  ],
5
+ "batch_norm_eps": 0.001,
6
+ "batch_norm_momentum": 0.99,
7
+ "depth_coefficient": 1.0,
8
+ "depth_divisor": 8,
9
+ "depthwise_padding": [],
10
+ "drop_connect_rate": 0.2,
11
+ "dropout_rate": 0.2,
12
  "dtype": "float32",
13
+ "expand_ratios": [
14
+ 1,
15
+ 6,
16
+ 6,
17
+ 6,
18
+ 6,
19
+ 6,
20
+ 6
21
+ ],
22
+ "hidden_act": "swish",
23
+ "hidden_dim": 1280,
24
  "id2label": {
25
  "0": "class_12_25",
26
  "1": "class_13_25",
 
28
  "3": "class_15_25",
29
  "4": "class_16_25"
30
  },
31
+ "image_size": 224,
32
+ "in_channels": [
33
+ 32,
34
+ 16,
35
+ 24,
36
+ 40,
37
+ 80,
38
+ 112,
39
+ 192
40
+ ],
41
  "initializer_range": 0.02,
42
+ "kernel_sizes": [
43
+ 3,
44
+ 3,
45
+ 5,
46
+ 3,
47
+ 5,
48
+ 5,
49
+ 3
50
+ ],
51
  "label2id": {
52
  "class_12_25": 0,
53
  "class_13_25": 1,
 
55
  "class_15_25": 3,
56
  "class_16_25": 4
57
  },
58
+ "model_type": "efficientnet",
59
+ "num_block_repeats": [
60
+ 1,
61
+ 2,
62
+ 2,
63
+ 3,
64
+ 3,
65
+ 4,
66
+ 1
67
+ ],
68
  "num_channels": 3,
69
+ "num_hidden_layers": 64,
70
+ "out_channels": [
71
+ 16,
72
+ 24,
73
+ 40,
74
+ 80,
75
+ 112,
76
+ 192,
77
+ 320
78
+ ],
79
+ "out_features": null,
80
+ "pooling_type": "mean",
81
  "problem_type": "single_label_classification",
82
+ "squeeze_expansion_ratio": 0.25,
83
+ "stage_names": [
84
+ "stem",
85
+ "stage1",
86
+ "stage2",
87
+ "stage3",
88
+ "stage4",
89
+ "stage5",
90
+ "stage6",
91
+ "stage7"
92
+ ],
93
+ "strides": [
94
+ 1,
95
+ 2,
96
+ 2,
97
+ 2,
98
+ 1,
99
+ 2,
100
+ 1
101
+ ],
102
+ "transformers_version": "4.57.1",
103
+ "width_coefficient": 1.0
104
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b0a12c470e16b68e486aa55de3bf27eecf508b74aed0251b0537ad3071a28b9d
3
- size 344400564
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:99820b02eedd6e490c575d54f2a29534f0b9cbacd578e76cb5180418ad22d110
3
+ size 16270500