mohammadsp99 commited on
Commit
b6b13fb
1 Parent(s): 232829f

End of training

Browse files
README.md CHANGED
@@ -1,3 +1,69 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: google/vit-base-patch16-224-in21k
4
+ tags:
5
+ - generated_from_trainer
6
+ metrics:
7
+ - accuracy
8
+ model-index:
9
+ - name: MyFoodModelViTFull
10
+ results: []
11
+ ---
12
+
13
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
14
+ should probably proofread and complete it, then remove this comment. -->
15
+
16
+ # MyFoodModelViTFull
17
+
18
+ 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.
19
+ It achieves the following results on the evaluation set:
20
+ - Loss: 0.8555
21
+ - Accuracy: 0.912
22
+
23
+ ## Model description
24
+
25
+ More information needed
26
+
27
+ ## Intended uses & limitations
28
+
29
+ More information needed
30
+
31
+ ## Training and evaluation data
32
+
33
+ More information needed
34
+
35
+ ## Training procedure
36
+
37
+ ### Training hyperparameters
38
+
39
+ The following hyperparameters were used during training:
40
+ - learning_rate: 5e-05
41
+ - train_batch_size: 16
42
+ - eval_batch_size: 16
43
+ - seed: 42
44
+ - gradient_accumulation_steps: 4
45
+ - total_train_batch_size: 64
46
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
+ - lr_scheduler_type: linear
48
+ - lr_scheduler_warmup_ratio: 0.1
49
+ - num_epochs: 6
50
+
51
+ ### Training results
52
+
53
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|
55
+ | 3.808 | 0.8 | 50 | 2.8728 | 0.799 |
56
+ | 2.2535 | 1.6 | 100 | 1.8121 | 0.877 |
57
+ | 1.5567 | 2.4 | 150 | 1.3607 | 0.905 |
58
+ | 1.1859 | 3.2 | 200 | 1.1252 | 0.906 |
59
+ | 0.9912 | 4.0 | 250 | 0.9753 | 0.915 |
60
+ | 0.8667 | 4.8 | 300 | 0.8917 | 0.919 |
61
+ | 0.7875 | 5.6 | 350 | 0.8555 | 0.912 |
62
+
63
+
64
+ ### Framework versions
65
+
66
+ - Transformers 4.41.0
67
+ - Pytorch 2.3.0+cu121
68
+ - Datasets 2.19.1
69
+ - Tokenizers 0.19.1
config.json ADDED
@@ -0,0 +1,230 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "google/vit-base-patch16-224-in21k",
3
+ "architectures": [
4
+ "ViTForImageClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "encoder_stride": 16,
8
+ "hidden_act": "gelu",
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",
222
+ "num_attention_heads": 12,
223
+ "num_channels": 3,
224
+ "num_hidden_layers": 12,
225
+ "patch_size": 16,
226
+ "problem_type": "single_label_classification",
227
+ "qkv_bias": true,
228
+ "torch_dtype": "float32",
229
+ "transformers_version": "4.41.0"
230
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:25fb425ff83c3d53295163800cb861f70d60ba4cfb55163b17d747f778f8f9d0
3
+ size 343528508
preprocessor_config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_valid_processor_keys": [
3
+ "images",
4
+ "do_resize",
5
+ "size",
6
+ "resample",
7
+ "do_rescale",
8
+ "rescale_factor",
9
+ "do_normalize",
10
+ "image_mean",
11
+ "image_std",
12
+ "return_tensors",
13
+ "data_format",
14
+ "input_data_format"
15
+ ],
16
+ "do_normalize": true,
17
+ "do_rescale": true,
18
+ "do_resize": true,
19
+ "image_mean": [
20
+ 0.5,
21
+ 0.5,
22
+ 0.5
23
+ ],
24
+ "image_processor_type": "ViTImageProcessor",
25
+ "image_std": [
26
+ 0.5,
27
+ 0.5,
28
+ 0.5
29
+ ],
30
+ "resample": 2,
31
+ "rescale_factor": 0.00392156862745098,
32
+ "size": {
33
+ "height": 224,
34
+ "width": 224
35
+ }
36
+ }
runs/May25_13-30-25_b77e6e7e4cf4/events.out.tfevents.1716643832.b77e6e7e4cf4.4071.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:373d64cb9c12b92342ac76442030276e04271bdf243da5a85d6b5c37ce3f1fa4
3
+ size 10273
runs/May25_13-30-25_b77e6e7e4cf4/events.out.tfevents.1716643893.b77e6e7e4cf4.4071.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:72f7fb6b0ac25f5573c39f9fe393ce5db707d0a23498e68e8b88618ac9d665cd
3
+ size 10322
runs/May25_13-32-23_b77e6e7e4cf4/events.out.tfevents.1716643947.b77e6e7e4cf4.4071.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bdfe0f632745c3246549788f98316ce59b17caa31ba213345f3308b44caffc7c
3
+ size 13870
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b4f5a870e705b361bda821bfc1b05474993e09edb6322e134db007679fc55c9b
3
+ size 5112