maazie commited on
Commit
de5c85e
·
1 Parent(s): e7cf2b4

Upload 2 files

Browse files
Files changed (2) hide show
  1. config.json +19 -16
  2. preprocessor_config.json +17 -17
config.json CHANGED
@@ -1,23 +1,22 @@
1
  {
2
  "architectures": [
3
- "EfficientNetForImageClassification"
4
  ],
5
  "depths": [
6
- 1,
7
- 1,
8
- 1,
9
- 1
 
 
 
 
 
 
 
 
 
10
  ],
11
-
12
- "num_classes": 1000,
13
- "width_coefficient": 1.0,
14
- "depth_coefficient": 1.0,
15
- "dropout_rate": 0.2,
16
- "image_size": 224,
17
- "mean": [0.485, 0.456, 0.406],
18
- "std": [0.229, 0.224, 0.225]
19
-
20
- }
21
  "id2label": {
22
  "0": "tench, Tinca tinca",
23
  "1": "goldfish, Carassius auratus",
@@ -2021,5 +2020,9 @@
2021
  "zebra": 340,
2022
  "zucchini, courgette": 939
2023
  },
2024
-
 
 
 
 
2025
  }
 
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",
 
2020
  "zebra": 340,
2021
  "zucchini, courgette": 939
2022
  },
2023
+ "layer_type": "bottleneck",
2024
+ "model_type": "resnet",
2025
+ "num_channels": 3,
2026
+ "torch_dtype": "float32",
2027
+ "transformers_version": "4.18.0.dev0"
2028
  }
preprocessor_config.json CHANGED
@@ -1,18 +1,18 @@
1
  {
2
- "crop_pct": 0.875,
3
- "do_normalize": true,
4
- "do_resize": true,
5
- "feature_extractor_type": "EfficientNetB0FeatureExtractor",
6
- "image_mean": [
7
- 0.485,
8
- 0.456,
9
- 0.406
10
- ],
11
- "image_std": [
12
- 0.229,
13
- 0.224,
14
- 0.225
15
- ],
16
- "resample": 3,
17
- "size": 224
18
- }
 
1
  {
2
+ "crop_pct": 0.875,
3
+ "do_normalize": true,
4
+ "do_resize": true,
5
+ "feature_extractor_type": "ConvNextFeatureExtractor",
6
+ "image_mean": [
7
+ 0.485,
8
+ 0.456,
9
+ 0.406
10
+ ],
11
+ "image_std": [
12
+ 0.229,
13
+ 0.224,
14
+ 0.225
15
+ ],
16
+ "resample": 3,
17
+ "size": 224
18
+ }