jacoballessio commited on
Commit
90d4635
1 Parent(s): 9716d1a

Upload 2 files

Browse files
Files changed (2) hide show
  1. config.json +3 -72
  2. model.safetensors +3 -0
config.json CHANGED
@@ -40,7 +40,7 @@
40
  "suppress_tokens": null,
41
  "begin_suppress_tokens": null,
42
  "architectures": [
43
- "EfficientNetForImageClassification"
44
  ],
45
  "finetuning_task": null,
46
  "id2label": {
@@ -62,75 +62,6 @@
62
  "problem_type": null,
63
  "_name_or_path": "",
64
  "transformers_version": "4.41.2",
65
- "num_channels": 3,
66
- "image_size": 600,
67
- "width_coefficient": 2.0,
68
- "depth_coefficient": 3.1,
69
- "depth_divisor": 8,
70
- "kernel_sizes": [
71
- 3,
72
- 3,
73
- 5,
74
- 3,
75
- 5,
76
- 5,
77
- 3
78
- ],
79
- "in_channels": [
80
- 32,
81
- 16,
82
- 24,
83
- 40,
84
- 80,
85
- 112,
86
- 192
87
- ],
88
- "out_channels": [
89
- 16,
90
- 24,
91
- 40,
92
- 80,
93
- 112,
94
- 192,
95
- 320
96
- ],
97
- "depthwise_padding": [],
98
- "strides": [
99
- 1,
100
- 2,
101
- 2,
102
- 2,
103
- 1,
104
- 2,
105
- 1
106
- ],
107
- "num_block_repeats": [
108
- 1,
109
- 2,
110
- 2,
111
- 3,
112
- 3,
113
- 4,
114
- 1
115
- ],
116
- "expand_ratios": [
117
- 1,
118
- 6,
119
- 6,
120
- 6,
121
- 6,
122
- 6,
123
- 6
124
- ],
125
- "squeeze_expansion_ratio": 0.25,
126
- "hidden_act": "swish",
127
- "hidden_dim": 2560,
128
- "pooling_type": "mean",
129
- "initializer_range": 0.02,
130
- "batch_norm_eps": 0.001,
131
- "batch_norm_momentum": 0.99,
132
- "dropout_rate": 0.5,
133
- "drop_connect_rate": 0.2,
134
- "num_hidden_layers": 64,
135
- "model_type": "efficientnet"
136
  }
 
40
  "suppress_tokens": null,
41
  "begin_suppress_tokens": null,
42
  "architectures": [
43
+ "CustomEfficientNet"
44
  ],
45
  "finetuning_task": null,
46
  "id2label": {
 
62
  "problem_type": null,
63
  "_name_or_path": "",
64
  "transformers_version": "4.41.2",
65
+ "num_classes": 2,
66
+ "model_type": "custom_efficientnet"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
67
  }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3d62df2da3dbe54aee172d22d5caecf0066f22c4524baa6a46db786ea62a5b67
3
+ size 16248480