arnabdhar commited on
Commit
95ba34b
1 Parent(s): 2b2bad5

End of training

Browse files
README.md ADDED
@@ -0,0 +1,78 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: other
3
+ base_model: google/mobilenet_v2_1.0_224
4
+ tags:
5
+ - pytoroch
6
+ - MobileNetV2ForImageClassification
7
+ - food-classification
8
+ - generated_from_trainer
9
+ metrics:
10
+ - accuracy
11
+ - recall
12
+ - precision
13
+ - f1
14
+ model-index:
15
+ - name: MobileNet-V2-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
+ # MobileNet-V2-food
23
+
24
+ This model is a fine-tuned version of [google/mobilenet_v2_1.0_224](https://huggingface.co/google/mobilenet_v2_1.0_224) on the ItsNotRohit/Food121-224 dataset.
25
+ It achieves the following results on the evaluation set:
26
+ - Loss: 1.6890
27
+ - Accuracy: 0.5793
28
+ - Recall: 0.5793
29
+ - Precision: 0.6006
30
+ - F1: 0.5769
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: 20329
53
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
54
+ - lr_scheduler_type: cosine
55
+ - training_steps: 20000
56
+
57
+ ### Training results
58
+
59
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | Recall | Precision | F1 |
60
+ |:-------------:|:-----:|:-----:|:---------------:|:--------:|:------:|:---------:|:------:|
61
+ | 2.9653 | 0.33 | 2000 | 2.7802 | 0.3438 | 0.3438 | 0.3932 | 0.3105 |
62
+ | 2.3854 | 0.66 | 4000 | 2.3105 | 0.4440 | 0.4440 | 0.4979 | 0.4336 |
63
+ | 2.1576 | 0.99 | 6000 | 2.0508 | 0.4958 | 0.4958 | 0.5263 | 0.4837 |
64
+ | 1.9767 | 1.32 | 8000 | 1.9860 | 0.5086 | 0.5086 | 0.5504 | 0.4956 |
65
+ | 1.9215 | 1.65 | 10000 | 1.8312 | 0.5462 | 0.5462 | 0.5815 | 0.5390 |
66
+ | 1.782 | 1.98 | 12000 | 1.8554 | 0.5441 | 0.5441 | 0.5864 | 0.5431 |
67
+ | 1.7755 | 2.31 | 14000 | 1.9241 | 0.5308 | 0.5308 | 0.5841 | 0.5272 |
68
+ | 1.7006 | 2.64 | 16000 | 1.8625 | 0.5451 | 0.5451 | 0.6004 | 0.5466 |
69
+ | 1.7289 | 2.98 | 18000 | 1.8560 | 0.5432 | 0.5432 | 0.5940 | 0.5395 |
70
+ | 1.7296 | 3.31 | 20000 | 1.6890 | 0.5793 | 0.5793 | 0.6006 | 0.5769 |
71
+
72
+
73
+ ### Framework versions
74
+
75
+ - Transformers 4.35.2
76
+ - Pytorch 2.1.0+cu121
77
+ - Datasets 2.15.0
78
+ - Tokenizers 0.15.0
config.json ADDED
@@ -0,0 +1,271 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "google/mobilenet_v2_1.0_224",
3
+ "architectures": [
4
+ "MobileNetV2ForImageClassification"
5
+ ],
6
+ "classifier_dropout_prob": 0.2,
7
+ "depth_divisible_by": 8,
8
+ "depth_multiplier": 1.0,
9
+ "expand_ratio": 6,
10
+ "finegrained_output": true,
11
+ "first_layer_is_expansion": true,
12
+ "hidden_act": "relu6",
13
+ "id2label": {
14
+ "0": "apple_pie",
15
+ "1": "baby_back_ribs",
16
+ "2": "baklava",
17
+ "3": "beef_carpaccio",
18
+ "4": "beef_tartare",
19
+ "5": "beet_salad",
20
+ "6": "beignets",
21
+ "7": "bibimbap",
22
+ "8": "biryani",
23
+ "9": "bread_pudding",
24
+ "10": "breakfast_burrito",
25
+ "11": "bruschetta",
26
+ "12": "caesar_salad",
27
+ "13": "cannoli",
28
+ "14": "caprese_salad",
29
+ "15": "carrot_cake",
30
+ "16": "ceviche",
31
+ "17": "chai",
32
+ "18": "chapati",
33
+ "19": "cheese_plate",
34
+ "20": "cheesecake",
35
+ "21": "chicken_curry",
36
+ "22": "chicken_quesadilla",
37
+ "23": "chicken_wings",
38
+ "24": "chocolate_cake",
39
+ "25": "chocolate_mousse",
40
+ "26": "chole_bhature",
41
+ "27": "churros",
42
+ "28": "clam_chowder",
43
+ "29": "club_sandwich",
44
+ "30": "crab_cakes",
45
+ "31": "creme_brulee",
46
+ "32": "croque_madame",
47
+ "33": "cup_cakes",
48
+ "34": "dabeli",
49
+ "35": "dal",
50
+ "36": "deviled_eggs",
51
+ "37": "dhokla",
52
+ "38": "donuts",
53
+ "39": "dosa",
54
+ "40": "dumplings",
55
+ "41": "edamame",
56
+ "42": "eggs_benedict",
57
+ "43": "escargots",
58
+ "44": "falafel",
59
+ "45": "filet_mignon",
60
+ "46": "fish_and_chips",
61
+ "47": "foie_gras",
62
+ "48": "french_fries",
63
+ "49": "french_onion_soup",
64
+ "50": "french_toast",
65
+ "51": "fried_calamari",
66
+ "52": "fried_rice",
67
+ "53": "frozen_yogurt",
68
+ "54": "garlic_bread",
69
+ "55": "gnocchi",
70
+ "56": "greek_salad",
71
+ "57": "grilled_cheese_sandwich",
72
+ "58": "grilled_salmon",
73
+ "59": "guacamole",
74
+ "60": "gyoza",
75
+ "61": "hamburger",
76
+ "62": "hot_and_sour_soup",
77
+ "63": "hot_dog",
78
+ "64": "huevos_rancheros",
79
+ "65": "hummus",
80
+ "66": "ice_cream",
81
+ "67": "idli",
82
+ "68": "jalebi",
83
+ "69": "kathi_rolls",
84
+ "70": "kofta",
85
+ "71": "kulfi",
86
+ "72": "lasagna",
87
+ "73": "lobster_bisque",
88
+ "74": "lobster_roll_sandwich",
89
+ "75": "macaroni_and_cheese",
90
+ "76": "macarons",
91
+ "77": "miso_soup",
92
+ "78": "momos",
93
+ "79": "mussels",
94
+ "80": "naan",
95
+ "81": "nachos",
96
+ "82": "omelette",
97
+ "83": "onion_rings",
98
+ "84": "oysters",
99
+ "85": "pad_thai",
100
+ "86": "paella",
101
+ "87": "pakoda",
102
+ "88": "pancakes",
103
+ "89": "pani_puri",
104
+ "90": "panna_cotta",
105
+ "91": "panner_butter_masala",
106
+ "92": "pav_bhaji",
107
+ "93": "peking_duck",
108
+ "94": "pho",
109
+ "95": "pizza",
110
+ "96": "pork_chop",
111
+ "97": "poutine",
112
+ "98": "prime_rib",
113
+ "99": "pulled_pork_sandwich",
114
+ "100": "ramen",
115
+ "101": "ravioli",
116
+ "102": "red_velvet_cake",
117
+ "103": "risotto",
118
+ "104": "samosa",
119
+ "105": "sashimi",
120
+ "106": "scallops",
121
+ "107": "seaweed_salad",
122
+ "108": "shrimp_and_grits",
123
+ "109": "spaghetti_bolognese",
124
+ "110": "spaghetti_carbonara",
125
+ "111": "spring_rolls",
126
+ "112": "steak",
127
+ "113": "strawberry_shortcake",
128
+ "114": "sushi",
129
+ "115": "tacos",
130
+ "116": "takoyaki",
131
+ "117": "tiramisu",
132
+ "118": "tuna_tartare",
133
+ "119": "vadapav",
134
+ "120": "waffles"
135
+ },
136
+ "image_size": 224,
137
+ "initializer_range": 0.02,
138
+ "label2id": {
139
+ "apple_pie": 0,
140
+ "baby_back_ribs": 1,
141
+ "baklava": 2,
142
+ "beef_carpaccio": 3,
143
+ "beef_tartare": 4,
144
+ "beet_salad": 5,
145
+ "beignets": 6,
146
+ "bibimbap": 7,
147
+ "biryani": 8,
148
+ "bread_pudding": 9,
149
+ "breakfast_burrito": 10,
150
+ "bruschetta": 11,
151
+ "caesar_salad": 12,
152
+ "cannoli": 13,
153
+ "caprese_salad": 14,
154
+ "carrot_cake": 15,
155
+ "ceviche": 16,
156
+ "chai": 17,
157
+ "chapati": 18,
158
+ "cheese_plate": 19,
159
+ "cheesecake": 20,
160
+ "chicken_curry": 21,
161
+ "chicken_quesadilla": 22,
162
+ "chicken_wings": 23,
163
+ "chocolate_cake": 24,
164
+ "chocolate_mousse": 25,
165
+ "chole_bhature": 26,
166
+ "churros": 27,
167
+ "clam_chowder": 28,
168
+ "club_sandwich": 29,
169
+ "crab_cakes": 30,
170
+ "creme_brulee": 31,
171
+ "croque_madame": 32,
172
+ "cup_cakes": 33,
173
+ "dabeli": 34,
174
+ "dal": 35,
175
+ "deviled_eggs": 36,
176
+ "dhokla": 37,
177
+ "donuts": 38,
178
+ "dosa": 39,
179
+ "dumplings": 40,
180
+ "edamame": 41,
181
+ "eggs_benedict": 42,
182
+ "escargots": 43,
183
+ "falafel": 44,
184
+ "filet_mignon": 45,
185
+ "fish_and_chips": 46,
186
+ "foie_gras": 47,
187
+ "french_fries": 48,
188
+ "french_onion_soup": 49,
189
+ "french_toast": 50,
190
+ "fried_calamari": 51,
191
+ "fried_rice": 52,
192
+ "frozen_yogurt": 53,
193
+ "garlic_bread": 54,
194
+ "gnocchi": 55,
195
+ "greek_salad": 56,
196
+ "grilled_cheese_sandwich": 57,
197
+ "grilled_salmon": 58,
198
+ "guacamole": 59,
199
+ "gyoza": 60,
200
+ "hamburger": 61,
201
+ "hot_and_sour_soup": 62,
202
+ "hot_dog": 63,
203
+ "huevos_rancheros": 64,
204
+ "hummus": 65,
205
+ "ice_cream": 66,
206
+ "idli": 67,
207
+ "jalebi": 68,
208
+ "kathi_rolls": 69,
209
+ "kofta": 70,
210
+ "kulfi": 71,
211
+ "lasagna": 72,
212
+ "lobster_bisque": 73,
213
+ "lobster_roll_sandwich": 74,
214
+ "macaroni_and_cheese": 75,
215
+ "macarons": 76,
216
+ "miso_soup": 77,
217
+ "momos": 78,
218
+ "mussels": 79,
219
+ "naan": 80,
220
+ "nachos": 81,
221
+ "omelette": 82,
222
+ "onion_rings": 83,
223
+ "oysters": 84,
224
+ "pad_thai": 85,
225
+ "paella": 86,
226
+ "pakoda": 87,
227
+ "pancakes": 88,
228
+ "pani_puri": 89,
229
+ "panna_cotta": 90,
230
+ "panner_butter_masala": 91,
231
+ "pav_bhaji": 92,
232
+ "peking_duck": 93,
233
+ "pho": 94,
234
+ "pizza": 95,
235
+ "pork_chop": 96,
236
+ "poutine": 97,
237
+ "prime_rib": 98,
238
+ "pulled_pork_sandwich": 99,
239
+ "ramen": 100,
240
+ "ravioli": 101,
241
+ "red_velvet_cake": 102,
242
+ "risotto": 103,
243
+ "samosa": 104,
244
+ "sashimi": 105,
245
+ "scallops": 106,
246
+ "seaweed_salad": 107,
247
+ "shrimp_and_grits": 108,
248
+ "spaghetti_bolognese": 109,
249
+ "spaghetti_carbonara": 110,
250
+ "spring_rolls": 111,
251
+ "steak": 112,
252
+ "strawberry_shortcake": 113,
253
+ "sushi": 114,
254
+ "tacos": 115,
255
+ "takoyaki": 116,
256
+ "tiramisu": 117,
257
+ "tuna_tartare": 118,
258
+ "vadapav": 119,
259
+ "waffles": 120
260
+ },
261
+ "layer_norm_eps": 0.001,
262
+ "min_depth": 8,
263
+ "model_type": "mobilenet_v2",
264
+ "num_channels": 3,
265
+ "output_stride": 32,
266
+ "problem_type": "single_label_classification",
267
+ "semantic_loss_ignore_index": 255,
268
+ "tf_padding": true,
269
+ "torch_dtype": "float32",
270
+ "transformers_version": "4.35.2"
271
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:135b7c60db45e8709cae754b1dc2ee12108d8f418222040726883bc77ef33788
3
+ size 9689972
preprocessor_config.json ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "crop_size": {
3
+ "height": 224,
4
+ "width": 224
5
+ },
6
+ "do_center_crop": true,
7
+ "do_normalize": true,
8
+ "do_rescale": true,
9
+ "do_resize": true,
10
+ "image_mean": [
11
+ 0.5,
12
+ 0.5,
13
+ 0.5
14
+ ],
15
+ "image_processor_type": "MobileNetV2ImageProcessor",
16
+ "image_std": [
17
+ 0.5,
18
+ 0.5,
19
+ 0.5
20
+ ],
21
+ "resample": 2,
22
+ "rescale_factor": 0.00392156862745098,
23
+ "size": {
24
+ "shortest_edge": 256
25
+ },
26
+ "use_square_size": false
27
+ }
runs/Dec20_15-51-05_28db14a188dc/events.out.tfevents.1703087481.28db14a188dc.265.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eab18cbe98af95bb4c089d8d9d188398303501c688b3e524af8e0db10754d952
3
+ size 21264
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2998430e0d310639d7919e0c541458c7bedffe926a79ec89d21895d128fce03d
3
+ size 4600