Hugging Face's logo

arieg
/
food

Image Classification
Transformers
TensorFlow
vit
generated_from_keras_callback
Model card Files
xet
 Community
arieg commited on
Commit
118d2fe
·
1 Parent(s): 50b16b1

End of training

Browse files
Files changed (3) hide show
  1. README.md +9 -9
  2. config.json +160 -202
  3. tf_model.h5 +2 -2
README.md CHANGED
@@ -15,9 +15,9 @@ probably proofread and complete it, then remove this comment. -->
15
 
16
  This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
- - Train Loss: 0.2861
19
- - Validation Loss: 0.2612
20
- - Train Accuracy: 0.929
21
  - Epoch: 4
22
 
23
  ## Model description
@@ -37,18 +37,18 @@ More information needed
37
  ### Training hyperparameters
38
 
39
  The following hyperparameters were used during training:
40
- - optimizer: {'name': 'AdamWeightDecay', 'clipnorm': 1.0, 'learning_rate': {'module': 'keras.optimizers.schedules', 'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 3e-05, 'decay_steps': 20000, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}, 'registered_name': None}, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False, 'weight_decay_rate': 0.01}
41
  - training_precision: float32
42
 
43
  ### Training results
44
 
45
  | Train Loss | Validation Loss | Train Accuracy | Epoch |
46
  |:----------:|:---------------:|:--------------:|:-----:|
47
- | 2.7216 | 1.6098 | 0.819 | 0 |
48
- | 1.1279 | 0.7218 | 0.899 | 1 |
49
- | 0.5720 | 0.4555 | 0.929 | 2 |
50
- | 0.3861 | 0.3125 | 0.923 | 3 |
51
- | 0.2861 | 0.2612 | 0.929 | 4 |
52
 
53
 
54
  ### Framework versions
 
15
 
16
  This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
+ - Train Loss: 2.2825
19
+ - Validation Loss: 2.2431
20
+ - Train Accuracy: 0.975
21
  - Epoch: 4
22
 
23
  ## Model description
 
37
  ### Training hyperparameters
38
 
39
  The following hyperparameters were used during training:
40
+ - optimizer: {'name': 'AdamWeightDecay', 'clipnorm': 1.0, 'learning_rate': {'module': 'keras.optimizers.schedules', 'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 3e-05, 'decay_steps': 7200, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}, 'registered_name': None}, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False, 'weight_decay_rate': 0.01}
41
  - training_precision: float32
42
 
43
  ### Training results
44
 
45
  | Train Loss | Validation Loss | Train Accuracy | Epoch |
46
  |:----------:|:---------------:|:--------------:|:-----:|
47
+ | 4.1953 | 3.9579 | 0.6562 | 0 |
48
+ | 3.6362 | 3.4187 | 0.925 | 1 |
49
+ | 3.1017 | 2.9427 | 0.9625 | 2 |
50
+ | 2.6561 | 2.5775 | 0.9688 | 3 |
51
+ | 2.2825 | 2.2431 | 0.975 | 4 |
52
 
53
 
54
  ### Framework versions
config.json CHANGED
@@ -9,213 +9,171 @@
9
  "hidden_dropout_prob": 0.0,
10
  "hidden_size": 768,
11
  "id2label": {
12
- "0": "apple_pie",
13
- "1": "baby_back_ribs",
14
- "10": "bruschetta",
15
- "100": "waffles",
16
- "11": "caesar_salad",
17
- "12": "cannoli",
18
- "13": "caprese_salad",
19
- "14": "carrot_cake",
20
- "15": "ceviche",
21
- "16": "cheesecake",
22
- "17": "cheese_plate",
23
- "18": "chicken_curry",
24
- "19": "chicken_quesadilla",
25
- "2": "baklava",
26
- "20": "chicken_wings",
27
- "21": "chocolate_cake",
28
- "22": "chocolate_mousse",
29
- "23": "churros",
30
- "24": "clam_chowder",
31
- "25": "club_sandwich",
32
- "26": "crab_cakes",
33
- "27": "creme_brulee",
34
- "28": "croque_madame",
35
- "29": "cup_cakes",
36
- "3": "beef_carpaccio",
37
- "30": "deviled_eggs",
38
- "31": "donuts",
39
- "32": "dumplings",
40
- "33": "edamame",
41
- "34": "eggs_benedict",
42
- "35": "escargots",
43
- "36": "falafel",
44
- "37": "filet_mignon",
45
- "38": "fish_and_chips",
46
- "39": "foie_gras",
47
- "4": "beef_tartare",
48
- "40": "french_fries",
49
- "41": "french_onion_soup",
50
- "42": "french_toast",
51
- "43": "fried_calamari",
52
- "44": "fried_rice",
53
- "45": "frozen_yogurt",
54
- "46": "garlic_bread",
55
- "47": "gnocchi",
56
- "48": "greek_salad",
57
- "49": "grilled_cheese_sandwich",
58
- "5": "beet_salad",
59
- "50": "grilled_salmon",
60
- "51": "guacamole",
61
- "52": "gyoza",
62
- "53": "hamburger",
63
- "54": "hot_and_sour_soup",
64
- "55": "hot_dog",
65
- "56": "huevos_rancheros",
66
- "57": "hummus",
67
- "58": "ice_cream",
68
- "59": "lasagna",
69
- "6": "beignets",
70
- "60": "lobster_bisque",
71
- "61": "lobster_roll_sandwich",
72
- "62": "macaroni_and_cheese",
73
- "63": "macarons",
74
- "64": "miso_soup",
75
- "65": "mussels",
76
- "66": "nachos",
77
- "67": "omelette",
78
- "68": "onion_rings",
79
- "69": "oysters",
80
- "7": "bibimbap",
81
- "70": "pad_thai",
82
- "71": "paella",
83
- "72": "pancakes",
84
- "73": "panna_cotta",
85
- "74": "peking_duck",
86
- "75": "pho",
87
- "76": "pizza",
88
- "77": "pork_chop",
89
- "78": "poutine",
90
- "79": "prime_rib",
91
- "8": "bread_pudding",
92
- "80": "pulled_pork_sandwich",
93
- "81": "ramen",
94
- "82": "ravioli",
95
- "83": "red_velvet_cake",
96
- "84": "risotto",
97
- "85": "samosa",
98
- "86": "sashimi",
99
- "87": "scallops",
100
- "88": "seaweed_salad",
101
- "89": "shrimp_and_grits",
102
- "9": "breakfast_burrito",
103
- "90": "spaghetti_bolognese",
104
- "91": "spaghetti_carbonara",
105
- "92": "spring_rolls",
106
- "93": "steak",
107
- "94": "strawberry_shortcake",
108
- "95": "sushi",
109
- "96": "tacos",
110
- "97": "takoyaki",
111
- "98": "tiramisu",
112
- "99": "tuna_tartare"
113
  },
114
  "image_size": 224,
115
  "initializer_range": 0.02,
116
  "intermediate_size": 3072,
117
  "label2id": {
118
- "apple_pie": "0",
119
- "baby_back_ribs": "1",
120
- "baklava": "2",
121
- "beef_carpaccio": "3",
122
- "beef_tartare": "4",
123
- "beet_salad": "5",
124
- "beignets": "6",
125
- "bibimbap": "7",
126
- "bread_pudding": "8",
127
- "breakfast_burrito": "9",
128
- "bruschetta": "10",
129
- "caesar_salad": "11",
130
- "cannoli": "12",
131
- "caprese_salad": "13",
132
- "carrot_cake": "14",
133
- "ceviche": "15",
134
- "cheese_plate": "17",
135
- "cheesecake": "16",
136
- "chicken_curry": "18",
137
- "chicken_quesadilla": "19",
138
- "chicken_wings": "20",
139
- "chocolate_cake": "21",
140
- "chocolate_mousse": "22",
141
- "churros": "23",
142
- "clam_chowder": "24",
143
- "club_sandwich": "25",
144
- "crab_cakes": "26",
145
- "creme_brulee": "27",
146
- "croque_madame": "28",
147
- "cup_cakes": "29",
148
- "deviled_eggs": "30",
149
- "donuts": "31",
150
- "dumplings": "32",
151
- "edamame": "33",
152
- "eggs_benedict": "34",
153
- "escargots": "35",
154
- "falafel": "36",
155
- "filet_mignon": "37",
156
- "fish_and_chips": "38",
157
- "foie_gras": "39",
158
- "french_fries": "40",
159
- "french_onion_soup": "41",
160
- "french_toast": "42",
161
- "fried_calamari": "43",
162
- "fried_rice": "44",
163
- "frozen_yogurt": "45",
164
- "garlic_bread": "46",
165
- "gnocchi": "47",
166
- "greek_salad": "48",
167
- "grilled_cheese_sandwich": "49",
168
- "grilled_salmon": "50",
169
- "guacamole": "51",
170
- "gyoza": "52",
171
- "hamburger": "53",
172
- "hot_and_sour_soup": "54",
173
- "hot_dog": "55",
174
- "huevos_rancheros": "56",
175
- "hummus": "57",
176
- "ice_cream": "58",
177
- "lasagna": "59",
178
- "lobster_bisque": "60",
179
- "lobster_roll_sandwich": "61",
180
- "macaroni_and_cheese": "62",
181
- "macarons": "63",
182
- "miso_soup": "64",
183
- "mussels": "65",
184
- "nachos": "66",
185
- "omelette": "67",
186
- "onion_rings": "68",
187
- "oysters": "69",
188
- "pad_thai": "70",
189
- "paella": "71",
190
- "pancakes": "72",
191
- "panna_cotta": "73",
192
- "peking_duck": "74",
193
- "pho": "75",
194
- "pizza": "76",
195
- "pork_chop": "77",
196
- "poutine": "78",
197
- "prime_rib": "79",
198
- "pulled_pork_sandwich": "80",
199
- "ramen": "81",
200
- "ravioli": "82",
201
- "red_velvet_cake": "83",
202
- "risotto": "84",
203
- "samosa": "85",
204
- "sashimi": "86",
205
- "scallops": "87",
206
- "seaweed_salad": "88",
207
- "shrimp_and_grits": "89",
208
- "spaghetti_bolognese": "90",
209
- "spaghetti_carbonara": "91",
210
- "spring_rolls": "92",
211
- "steak": "93",
212
- "strawberry_shortcake": "94",
213
- "sushi": "95",
214
- "tacos": "96",
215
- "takoyaki": "97",
216
- "tiramisu": "98",
217
- "tuna_tartare": "99",
218
- "waffles": "100"
219
  },
220
  "layer_norm_eps": 1e-12,
221
  "model_type": "vit",
 
9
  "hidden_dropout_prob": 0.0,
10
  "hidden_size": 768,
11
  "id2label": {
12
+ "0": "56248",
13
+ "1": "56249",
14
+ "10": "56470",
15
+ "11": "56471",
16
+ "12": "56472",
17
+ "13": "56474",
18
+ "14": "56493",
19
+ "15": "56495",
20
+ "16": "56496",
21
+ "17": "56497",
22
+ "18": "56498",
23
+ "19": "56499",
24
+ "2": "56273",
25
+ "20": "56516",
26
+ "21": "56517",
27
+ "22": "56518",
28
+ "23": "56519",
29
+ "24": "56520",
30
+ "25": "56521",
31
+ "26": "56639",
32
+ "27": "56640",
33
+ "28": "56641",
34
+ "29": "56645",
35
+ "3": "56274",
36
+ "30": "56646",
37
+ "31": "56648",
38
+ "32": "56649",
39
+ "33": "56650",
40
+ "34": "56651",
41
+ "35": "56686",
42
+ "36": "56687",
43
+ "37": "56688",
44
+ "38": "56689",
45
+ "39": "56690",
46
+ "4": "56275",
47
+ "40": "56691",
48
+ "41": "56692",
49
+ "42": "56693",
50
+ "43": "56694",
51
+ "44": "56695",
52
+ "45": "56696",
53
+ "46": "56795",
54
+ "47": "56796",
55
+ "48": "56797",
56
+ "49": "56798",
57
+ "5": "56465",
58
+ "50": "56799",
59
+ "51": "56800",
60
+ "52": "56801",
61
+ "53": "56802",
62
+ "54": "56803",
63
+ "55": "56804",
64
+ "56": "56805",
65
+ "57": "56888",
66
+ "58": "57164",
67
+ "59": "57175",
68
+ "6": "56466",
69
+ "60": "57176",
70
+ "61": "57177",
71
+ "62": "57178",
72
+ "63": "57179",
73
+ "64": "57180",
74
+ "65": "57344",
75
+ "66": "57360",
76
+ "67": "57371",
77
+ "68": "57417",
78
+ "69": "57418",
79
+ "7": "56467",
80
+ "70": "57440",
81
+ "71": "57442",
82
+ "72": "57500",
83
+ "73": "57569",
84
+ "74": "57626",
85
+ "75": "57627",
86
+ "76": "57628",
87
+ "77": "57629",
88
+ "78": "57630",
89
+ "79": "57639",
90
+ "8": "56468",
91
+ "9": "56469"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
92
  },
93
  "image_size": 224,
94
  "initializer_range": 0.02,
95
  "intermediate_size": 3072,
96
  "label2id": {
97
+ "56248": "0",
98
+ "56249": "1",
99
+ "56273": "2",
100
+ "56274": "3",
101
+ "56275": "4",
102
+ "56465": "5",
103
+ "56466": "6",
104
+ "56467": "7",
105
+ "56468": "8",
106
+ "56469": "9",
107
+ "56470": "10",
108
+ "56471": "11",
109
+ "56472": "12",
110
+ "56474": "13",
111
+ "56493": "14",
112
+ "56495": "15",
113
+ "56496": "16",
114
+ "56497": "17",
115
+ "56498": "18",
116
+ "56499": "19",
117
+ "56516": "20",
118
+ "56517": "21",
119
+ "56518": "22",
120
+ "56519": "23",
121
+ "56520": "24",
122
+ "56521": "25",
123
+ "56639": "26",
124
+ "56640": "27",
125
+ "56641": "28",
126
+ "56645": "29",
127
+ "56646": "30",
128
+ "56648": "31",
129
+ "56649": "32",
130
+ "56650": "33",
131
+ "56651": "34",
132
+ "56686": "35",
133
+ "56687": "36",
134
+ "56688": "37",
135
+ "56689": "38",
136
+ "56690": "39",
137
+ "56691": "40",
138
+ "56692": "41",
139
+ "56693": "42",
140
+ "56694": "43",
141
+ "56695": "44",
142
+ "56696": "45",
143
+ "56795": "46",
144
+ "56796": "47",
145
+ "56797": "48",
146
+ "56798": "49",
147
+ "56799": "50",
148
+ "56800": "51",
149
+ "56801": "52",
150
+ "56802": "53",
151
+ "56803": "54",
152
+ "56804": "55",
153
+ "56805": "56",
154
+ "56888": "57",
155
+ "57164": "58",
156
+ "57175": "59",
157
+ "57176": "60",
158
+ "57177": "61",
159
+ "57178": "62",
160
+ "57179": "63",
161
+ "57180": "64",
162
+ "57344": "65",
163
+ "57360": "66",
164
+ "57371": "67",
165
+ "57417": "68",
166
+ "57418": "69",
167
+ "57440": "70",
168
+ "57442": "71",
169
+ "57500": "72",
170
+ "57569": "73",
171
+ "57626": "74",
172
+ "57627": "75",
173
+ "57628": "76",
174
+ "57629": "77",
175
+ "57630": "78",
176
+ "57639": "79"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
177
  },
178
  "layer_norm_eps": 1e-12,
179
  "model_type": "vit",
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e159370e181fbc13fd111975891b6dd3cf5ea2e2cf57cdbe1b21b35edbbd42c1
3
- size 343790264
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8bcb0be5c078d818d2085cd3ef225fe498679fc4861d82aa543db711be6bfccd
3
+ size 343725752