arnabdhar commited on
Commit
8868ce7
1 Parent(s): a73f695

End of training

Browse files
README.md ADDED
@@ -0,0 +1,79 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: microsoft/swinv2-base-patch4-window8-256
4
+ tags:
5
+ - pytoroch
6
+ - Swinv2ForImageClassification
7
+ - food-classification
8
+ - generated_from_trainer
9
+ metrics:
10
+ - accuracy
11
+ - recall
12
+ - precision
13
+ - f1
14
+ model-index:
15
+ - name: Swin-V2-base-Food
16
+ results: []
17
+ ---
18
+
19
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
20
+ should probably proofread and complete it, then remove this comment. -->
21
+
22
+ # Swin-V2-base-Food
23
+
24
+ This model is a fine-tuned version of [microsoft/swinv2-base-patch4-window8-256](https://huggingface.co/microsoft/swinv2-base-patch4-window8-256) on the ItsNotRohit/Food121-224 dataset.
25
+ It achieves the following results on the evaluation set:
26
+ - Loss: 0.7099
27
+ - Accuracy: 0.8160
28
+ - Recall: 0.8160
29
+ - Precision: 0.8168
30
+ - F1: 0.8159
31
+
32
+ ## Model description
33
+
34
+ More information needed
35
+
36
+ ## Intended uses & limitations
37
+
38
+ More information needed
39
+
40
+ ## Training and evaluation data
41
+
42
+ More information needed
43
+
44
+ ## Training procedure
45
+
46
+ ### Training hyperparameters
47
+
48
+ The following hyperparameters were used during training:
49
+ - learning_rate: 5e-05
50
+ - train_batch_size: 16
51
+ - eval_batch_size: 128
52
+ - seed: 17769929
53
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
54
+ - lr_scheduler_type: cosine
55
+ - lr_scheduler_warmup_ratio: 0.01
56
+ - training_steps: 20000
57
+
58
+ ### Training results
59
+
60
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | Recall | Precision | F1 |
61
+ |:-------------:|:-----:|:-----:|:---------------:|:--------:|:------:|:---------:|:------:|
62
+ | 1.5169 | 0.33 | 2000 | 1.2680 | 0.6746 | 0.6746 | 0.7019 | 0.6737 |
63
+ | 1.2362 | 0.66 | 4000 | 1.0759 | 0.7169 | 0.7169 | 0.7411 | 0.7178 |
64
+ | 1.1076 | 0.99 | 6000 | 0.9757 | 0.7437 | 0.7437 | 0.7593 | 0.7430 |
65
+ | 0.9163 | 1.32 | 8000 | 0.9123 | 0.7623 | 0.7623 | 0.7737 | 0.7628 |
66
+ | 0.8291 | 1.65 | 10000 | 0.8397 | 0.7807 | 0.7807 | 0.7874 | 0.7796 |
67
+ | 0.7949 | 1.98 | 12000 | 0.7724 | 0.7965 | 0.7965 | 0.8014 | 0.7965 |
68
+ | 0.6455 | 2.31 | 14000 | 0.7458 | 0.8030 | 0.8030 | 0.8069 | 0.8031 |
69
+ | 0.6332 | 2.64 | 16000 | 0.7222 | 0.8110 | 0.8110 | 0.8122 | 0.8106 |
70
+ | 0.6132 | 2.98 | 18000 | 0.7021 | 0.8154 | 0.8154 | 0.8170 | 0.8155 |
71
+ | 0.57 | 3.31 | 20000 | 0.7099 | 0.8160 | 0.8160 | 0.8168 | 0.8159 |
72
+
73
+
74
+ ### Framework versions
75
+
76
+ - Transformers 4.35.2
77
+ - Pytorch 2.1.0+cu121
78
+ - Datasets 2.15.0
79
+ - Tokenizers 0.15.0
config.json ADDED
@@ -0,0 +1,292 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "microsoft/swinv2-base-patch4-window8-256",
3
+ "architectures": [
4
+ "Swinv2ForImageClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "depths": [
8
+ 2,
9
+ 2,
10
+ 18,
11
+ 2
12
+ ],
13
+ "drop_path_rate": 0.1,
14
+ "embed_dim": 128,
15
+ "encoder_stride": 32,
16
+ "hidden_act": "gelu",
17
+ "hidden_dropout_prob": 0.0,
18
+ "hidden_size": 1024,
19
+ "id2label": {
20
+ "0": "apple_pie",
21
+ "1": "baby_back_ribs",
22
+ "2": "baklava",
23
+ "3": "beef_carpaccio",
24
+ "4": "beef_tartare",
25
+ "5": "beet_salad",
26
+ "6": "beignets",
27
+ "7": "bibimbap",
28
+ "8": "biryani",
29
+ "9": "bread_pudding",
30
+ "10": "breakfast_burrito",
31
+ "11": "bruschetta",
32
+ "12": "caesar_salad",
33
+ "13": "cannoli",
34
+ "14": "caprese_salad",
35
+ "15": "carrot_cake",
36
+ "16": "ceviche",
37
+ "17": "chai",
38
+ "18": "chapati",
39
+ "19": "cheese_plate",
40
+ "20": "cheesecake",
41
+ "21": "chicken_curry",
42
+ "22": "chicken_quesadilla",
43
+ "23": "chicken_wings",
44
+ "24": "chocolate_cake",
45
+ "25": "chocolate_mousse",
46
+ "26": "chole_bhature",
47
+ "27": "churros",
48
+ "28": "clam_chowder",
49
+ "29": "club_sandwich",
50
+ "30": "crab_cakes",
51
+ "31": "creme_brulee",
52
+ "32": "croque_madame",
53
+ "33": "cup_cakes",
54
+ "34": "dabeli",
55
+ "35": "dal",
56
+ "36": "deviled_eggs",
57
+ "37": "dhokla",
58
+ "38": "donuts",
59
+ "39": "dosa",
60
+ "40": "dumplings",
61
+ "41": "edamame",
62
+ "42": "eggs_benedict",
63
+ "43": "escargots",
64
+ "44": "falafel",
65
+ "45": "filet_mignon",
66
+ "46": "fish_and_chips",
67
+ "47": "foie_gras",
68
+ "48": "french_fries",
69
+ "49": "french_onion_soup",
70
+ "50": "french_toast",
71
+ "51": "fried_calamari",
72
+ "52": "fried_rice",
73
+ "53": "frozen_yogurt",
74
+ "54": "garlic_bread",
75
+ "55": "gnocchi",
76
+ "56": "greek_salad",
77
+ "57": "grilled_cheese_sandwich",
78
+ "58": "grilled_salmon",
79
+ "59": "guacamole",
80
+ "60": "gyoza",
81
+ "61": "hamburger",
82
+ "62": "hot_and_sour_soup",
83
+ "63": "hot_dog",
84
+ "64": "huevos_rancheros",
85
+ "65": "hummus",
86
+ "66": "ice_cream",
87
+ "67": "idli",
88
+ "68": "jalebi",
89
+ "69": "kathi_rolls",
90
+ "70": "kofta",
91
+ "71": "kulfi",
92
+ "72": "lasagna",
93
+ "73": "lobster_bisque",
94
+ "74": "lobster_roll_sandwich",
95
+ "75": "macaroni_and_cheese",
96
+ "76": "macarons",
97
+ "77": "miso_soup",
98
+ "78": "momos",
99
+ "79": "mussels",
100
+ "80": "naan",
101
+ "81": "nachos",
102
+ "82": "omelette",
103
+ "83": "onion_rings",
104
+ "84": "oysters",
105
+ "85": "pad_thai",
106
+ "86": "paella",
107
+ "87": "pakoda",
108
+ "88": "pancakes",
109
+ "89": "pani_puri",
110
+ "90": "panna_cotta",
111
+ "91": "panner_butter_masala",
112
+ "92": "pav_bhaji",
113
+ "93": "peking_duck",
114
+ "94": "pho",
115
+ "95": "pizza",
116
+ "96": "pork_chop",
117
+ "97": "poutine",
118
+ "98": "prime_rib",
119
+ "99": "pulled_pork_sandwich",
120
+ "100": "ramen",
121
+ "101": "ravioli",
122
+ "102": "red_velvet_cake",
123
+ "103": "risotto",
124
+ "104": "samosa",
125
+ "105": "sashimi",
126
+ "106": "scallops",
127
+ "107": "seaweed_salad",
128
+ "108": "shrimp_and_grits",
129
+ "109": "spaghetti_bolognese",
130
+ "110": "spaghetti_carbonara",
131
+ "111": "spring_rolls",
132
+ "112": "steak",
133
+ "113": "strawberry_shortcake",
134
+ "114": "sushi",
135
+ "115": "tacos",
136
+ "116": "takoyaki",
137
+ "117": "tiramisu",
138
+ "118": "tuna_tartare",
139
+ "119": "vadapav",
140
+ "120": "waffles"
141
+ },
142
+ "image_size": 256,
143
+ "initializer_range": 0.02,
144
+ "label2id": {
145
+ "apple_pie": 0,
146
+ "baby_back_ribs": 1,
147
+ "baklava": 2,
148
+ "beef_carpaccio": 3,
149
+ "beef_tartare": 4,
150
+ "beet_salad": 5,
151
+ "beignets": 6,
152
+ "bibimbap": 7,
153
+ "biryani": 8,
154
+ "bread_pudding": 9,
155
+ "breakfast_burrito": 10,
156
+ "bruschetta": 11,
157
+ "caesar_salad": 12,
158
+ "cannoli": 13,
159
+ "caprese_salad": 14,
160
+ "carrot_cake": 15,
161
+ "ceviche": 16,
162
+ "chai": 17,
163
+ "chapati": 18,
164
+ "cheese_plate": 19,
165
+ "cheesecake": 20,
166
+ "chicken_curry": 21,
167
+ "chicken_quesadilla": 22,
168
+ "chicken_wings": 23,
169
+ "chocolate_cake": 24,
170
+ "chocolate_mousse": 25,
171
+ "chole_bhature": 26,
172
+ "churros": 27,
173
+ "clam_chowder": 28,
174
+ "club_sandwich": 29,
175
+ "crab_cakes": 30,
176
+ "creme_brulee": 31,
177
+ "croque_madame": 32,
178
+ "cup_cakes": 33,
179
+ "dabeli": 34,
180
+ "dal": 35,
181
+ "deviled_eggs": 36,
182
+ "dhokla": 37,
183
+ "donuts": 38,
184
+ "dosa": 39,
185
+ "dumplings": 40,
186
+ "edamame": 41,
187
+ "eggs_benedict": 42,
188
+ "escargots": 43,
189
+ "falafel": 44,
190
+ "filet_mignon": 45,
191
+ "fish_and_chips": 46,
192
+ "foie_gras": 47,
193
+ "french_fries": 48,
194
+ "french_onion_soup": 49,
195
+ "french_toast": 50,
196
+ "fried_calamari": 51,
197
+ "fried_rice": 52,
198
+ "frozen_yogurt": 53,
199
+ "garlic_bread": 54,
200
+ "gnocchi": 55,
201
+ "greek_salad": 56,
202
+ "grilled_cheese_sandwich": 57,
203
+ "grilled_salmon": 58,
204
+ "guacamole": 59,
205
+ "gyoza": 60,
206
+ "hamburger": 61,
207
+ "hot_and_sour_soup": 62,
208
+ "hot_dog": 63,
209
+ "huevos_rancheros": 64,
210
+ "hummus": 65,
211
+ "ice_cream": 66,
212
+ "idli": 67,
213
+ "jalebi": 68,
214
+ "kathi_rolls": 69,
215
+ "kofta": 70,
216
+ "kulfi": 71,
217
+ "lasagna": 72,
218
+ "lobster_bisque": 73,
219
+ "lobster_roll_sandwich": 74,
220
+ "macaroni_and_cheese": 75,
221
+ "macarons": 76,
222
+ "miso_soup": 77,
223
+ "momos": 78,
224
+ "mussels": 79,
225
+ "naan": 80,
226
+ "nachos": 81,
227
+ "omelette": 82,
228
+ "onion_rings": 83,
229
+ "oysters": 84,
230
+ "pad_thai": 85,
231
+ "paella": 86,
232
+ "pakoda": 87,
233
+ "pancakes": 88,
234
+ "pani_puri": 89,
235
+ "panna_cotta": 90,
236
+ "panner_butter_masala": 91,
237
+ "pav_bhaji": 92,
238
+ "peking_duck": 93,
239
+ "pho": 94,
240
+ "pizza": 95,
241
+ "pork_chop": 96,
242
+ "poutine": 97,
243
+ "prime_rib": 98,
244
+ "pulled_pork_sandwich": 99,
245
+ "ramen": 100,
246
+ "ravioli": 101,
247
+ "red_velvet_cake": 102,
248
+ "risotto": 103,
249
+ "samosa": 104,
250
+ "sashimi": 105,
251
+ "scallops": 106,
252
+ "seaweed_salad": 107,
253
+ "shrimp_and_grits": 108,
254
+ "spaghetti_bolognese": 109,
255
+ "spaghetti_carbonara": 110,
256
+ "spring_rolls": 111,
257
+ "steak": 112,
258
+ "strawberry_shortcake": 113,
259
+ "sushi": 114,
260
+ "tacos": 115,
261
+ "takoyaki": 116,
262
+ "tiramisu": 117,
263
+ "tuna_tartare": 118,
264
+ "vadapav": 119,
265
+ "waffles": 120
266
+ },
267
+ "layer_norm_eps": 1e-05,
268
+ "mlp_ratio": 4.0,
269
+ "model_type": "swinv2",
270
+ "num_channels": 3,
271
+ "num_heads": [
272
+ 4,
273
+ 8,
274
+ 16,
275
+ 32
276
+ ],
277
+ "num_layers": 4,
278
+ "patch_size": 4,
279
+ "path_norm": true,
280
+ "pretrained_window_sizes": [
281
+ 0,
282
+ 0,
283
+ 0,
284
+ 0
285
+ ],
286
+ "problem_type": "single_label_classification",
287
+ "qkv_bias": true,
288
+ "torch_dtype": "float32",
289
+ "transformers_version": "4.35.2",
290
+ "use_absolute_embeddings": false,
291
+ "window_size": 8
292
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5b495e316e8335a1b6b97f808a572c3cd5ad5b66dca801d93b257898c3daf031
3
+ size 348133436
preprocessor_config.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "do_rescale": true,
4
+ "do_resize": true,
5
+ "image_mean": [
6
+ 0.485,
7
+ 0.456,
8
+ 0.406
9
+ ],
10
+ "image_processor_type": "ViTImageProcessor",
11
+ "image_std": [
12
+ 0.229,
13
+ 0.224,
14
+ 0.225
15
+ ],
16
+ "resample": 3,
17
+ "rescale_factor": 0.00392156862745098,
18
+ "size": {
19
+ "height": 256,
20
+ "width": 256
21
+ }
22
+ }
runs/Dec21_13-55-09_fdbe9f385732/events.out.tfevents.1703166920.fdbe9f385732.212.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:95a46612813c4314592b19921d084917980e78079070a7811cb472f12f142d6d
3
+ size 21473
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5f0fe3194b0092e09185386c26999b6dc3c722fc08df0cb77df9817528db6078
3
+ size 4600