shingguy1 commited on
Commit
92c9fec
·
verified ·
1 Parent(s): db70509

Upload ConvNextForImageClassification

Browse files
Files changed (2) hide show
  1. config.json +4 -5
  2. model.safetensors +1 -1
config.json CHANGED
@@ -20,8 +20,8 @@
20
  "0": "pizza",
21
  "1": "hamburger",
22
  "2": "sushi",
23
- "3": "salad",
24
- "4": "pasta",
25
  "5": "ice_cream",
26
  "6": "fried_rice",
27
  "7": "tacos",
@@ -31,13 +31,13 @@
31
  "image_size": 224,
32
  "initializer_range": 0.02,
33
  "label2id": {
 
34
  "chocolate_cake": 9,
35
  "fried_rice": 6,
36
  "hamburger": 1,
37
  "ice_cream": 5,
38
- "pasta": 4,
39
  "pizza": 0,
40
- "salad": 3,
41
  "steak": 8,
42
  "sushi": 2,
43
  "tacos": 7
@@ -54,7 +54,6 @@
54
  4
55
  ],
56
  "patch_size": 4,
57
- "problem_type": "single_label_classification",
58
  "stage_names": [
59
  "stem",
60
  "stage1",
 
20
  "0": "pizza",
21
  "1": "hamburger",
22
  "2": "sushi",
23
+ "3": "caesar_salad",
24
+ "4": "spaghetti_bolognese",
25
  "5": "ice_cream",
26
  "6": "fried_rice",
27
  "7": "tacos",
 
31
  "image_size": 224,
32
  "initializer_range": 0.02,
33
  "label2id": {
34
+ "caesar_salad": 3,
35
  "chocolate_cake": 9,
36
  "fried_rice": 6,
37
  "hamburger": 1,
38
  "ice_cream": 5,
 
39
  "pizza": 0,
40
+ "spaghetti_bolognese": 4,
41
  "steak": 8,
42
  "sushi": 2,
43
  "tacos": 7
 
54
  4
55
  ],
56
  "patch_size": 4,
 
57
  "stage_names": [
58
  "stem",
59
  "stage1",
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5a94b486f154e6bbc41f1942d370e7139f6acf84a2c2e5d6ec40c00bbebd98c6
3
  size 111332552
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f31edfeb2971c45dfc787356ac2dc8b818b7e0ade443d688df7b978365709ead
3
  size 111332552