End of training
Browse files- README.md +79 -0
- config.json +292 -0
- model.safetensors +3 -0
- preprocessor_config.json +22 -0
- runs/Dec21_13-55-09_fdbe9f385732/events.out.tfevents.1703166920.fdbe9f385732.212.0 +3 -0
- training_args.bin +3 -0
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
|