sngsfydy commited on
Commit
9dea262
·
1 Parent(s): 3a381b2

Upload EfficientNetForImageClassification

Browse files
Files changed (2) hide show
  1. config.json +98 -0
  2. pytorch_model.bin +3 -0
config.json ADDED
@@ -0,0 +1,98 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "dennisjooo/Birds-Classifier-EfficientNetB2",
3
+ "architectures": [
4
+ "EfficientNetForImageClassification"
5
+ ],
6
+ "batch_norm_eps": 0.001,
7
+ "batch_norm_momentum": 0.99,
8
+ "depth_coefficient": 1.2,
9
+ "depth_divisor": 8,
10
+ "depthwise_padding": [
11
+ 5,
12
+ 8,
13
+ 16
14
+ ],
15
+ "drop_connect_rate": 0.2,
16
+ "dropout_rate": 0.3,
17
+ "expand_ratios": [
18
+ 1,
19
+ 6,
20
+ 6,
21
+ 6,
22
+ 6,
23
+ 6,
24
+ 6
25
+ ],
26
+ "hidden_act": "swish",
27
+ "hidden_dim": 1408,
28
+ "id2label": {
29
+ "0": "0",
30
+ "1": "1",
31
+ "2": "2",
32
+ "3": "3",
33
+ "4": "4"
34
+ },
35
+ "image_size": 260,
36
+ "in_channels": [
37
+ 32,
38
+ 16,
39
+ 24,
40
+ 40,
41
+ 80,
42
+ 112,
43
+ 192
44
+ ],
45
+ "initializer_range": 0.02,
46
+ "kernel_sizes": [
47
+ 3,
48
+ 3,
49
+ 5,
50
+ 3,
51
+ 5,
52
+ 5,
53
+ 3
54
+ ],
55
+ "label2id": {
56
+ "0": 0,
57
+ "1": 1,
58
+ "2": 2,
59
+ "3": 3,
60
+ "4": 4
61
+ },
62
+ "model_type": "efficientnet",
63
+ "num_block_repeats": [
64
+ 1,
65
+ 2,
66
+ 2,
67
+ 3,
68
+ 3,
69
+ 4,
70
+ 1
71
+ ],
72
+ "num_channels": 3,
73
+ "num_hidden_layers": 64,
74
+ "out_channels": [
75
+ 16,
76
+ 24,
77
+ 40,
78
+ 80,
79
+ 112,
80
+ 192,
81
+ 320
82
+ ],
83
+ "pooling_type": "mean",
84
+ "problem_type": "single_label_classification",
85
+ "squeeze_expansion_ratio": 0.25,
86
+ "strides": [
87
+ 1,
88
+ 2,
89
+ 2,
90
+ 2,
91
+ 1,
92
+ 2,
93
+ 1
94
+ ],
95
+ "torch_dtype": "float32",
96
+ "transformers_version": "4.34.1",
97
+ "width_coefficient": 1.1
98
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cf3da156f79c229277771f6c113c9a8076ab7ff0991331d3cb60ed8a847f97df
3
+ size 31281658