maazie commited on
Commit
326ced3
·
1 Parent(s): 09973cc

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +12 -15
config.json CHANGED
@@ -1,25 +1,22 @@
1
  {
2
  "architectures": [
3
- "EfficientNetForImageClassification"
4
  ],
5
  "depths": [
6
- 1.0,
7
- 1.0,
8
- 1.0,
9
- 1.0,
10
- 1.0,
11
- 2.0,
12
- 2.0,
13
- 2.0
14
  ],
15
- "downsample_in_first_stage": true,
16
- "embedding_size": 1280,
17
- "hidden_act": "swish",
18
  "hidden_sizes": [
19
- 320,
20
- 1280,
 
21
  2048
22
- ]
23
  "id2label": {
24
  "0": "tench, Tinca tinca",
25
  "1": "goldfish, Carassius auratus",
 
1
  {
2
  "architectures": [
3
+ "ResNetForImageClassification"
4
  ],
5
  "depths": [
6
+ 3,
7
+ 4,
8
+ 6,
9
+ 3
 
 
 
 
10
  ],
11
+ "downsample_in_first_stage": false,
12
+ "embedding_size": 64,
13
+ "hidden_act": "relu",
14
  "hidden_sizes": [
15
+ 256,
16
+ 512,
17
+ 1024,
18
  2048
19
+ ],
20
  "id2label": {
21
  "0": "tench, Tinca tinca",
22
  "1": "goldfish, Carassius auratus",