nst-t commited on
Commit
0587c99
1 Parent(s): fa9d180

End of training

Browse files
Files changed (4) hide show
  1. README.md +60 -0
  2. config.json +228 -0
  3. preprocessor_config.json +22 -0
  4. tf_model.h5 +3 -0
README.md ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: google/vit-base-patch16-224-in21k
5
+ tags:
6
+ - generated_from_keras_callback
7
+ model-index:
8
+ - name: nst-t/food_classifier
9
+ results: []
10
+ ---
11
+
12
+ <!-- This model card has been generated automatically according to the information Keras had access to. You should
13
+ probably proofread and complete it, then remove this comment. -->
14
+
15
+ # nst-t/food_classifier
16
+
17
+ 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.
18
+ It achieves the following results on the evaluation set:
19
+ - Train Loss: 0.3636
20
+ - Validation Loss: 0.3328
21
+ - Train Accuracy: 0.922
22
+ - Epoch: 4
23
+
24
+ ## Model description
25
+
26
+ More information needed
27
+
28
+ ## Intended uses & limitations
29
+
30
+ More information needed
31
+
32
+ ## Training and evaluation data
33
+
34
+ More information needed
35
+
36
+ ## Training procedure
37
+
38
+ ### Training hyperparameters
39
+
40
+ The following hyperparameters were used during training:
41
+ - optimizer: {'name': 'AdamWeightDecay', '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}
42
+ - training_precision: float32
43
+
44
+ ### Training results
45
+
46
+ | Train Loss | Validation Loss | Train Accuracy | Epoch |
47
+ |:----------:|:---------------:|:--------------:|:-----:|
48
+ | 2.8054 | 1.6563 | 0.819 | 0 |
49
+ | 1.2309 | 0.8322 | 0.879 | 1 |
50
+ | 0.6857 | 0.5211 | 0.913 | 2 |
51
+ | 0.4820 | 0.4088 | 0.911 | 3 |
52
+ | 0.3636 | 0.3328 | 0.922 | 4 |
53
+
54
+
55
+ ### Framework versions
56
+
57
+ - Transformers 4.44.2
58
+ - TensorFlow 2.17.0
59
+ - Datasets 3.0.0
60
+ - Tokenizers 0.19.1
config.json ADDED
@@ -0,0 +1,228 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ "qkv_bias": true,
227
+ "transformers_version": "4.44.2"
228
+ }
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.5,
7
+ 0.5,
8
+ 0.5
9
+ ],
10
+ "image_processor_type": "ViTImageProcessor",
11
+ "image_std": [
12
+ 0.5,
13
+ 0.5,
14
+ 0.5
15
+ ],
16
+ "resample": 2,
17
+ "rescale_factor": 0.00392156862745098,
18
+ "size": {
19
+ "height": 224,
20
+ "width": 224
21
+ }
22
+ }
tf_model.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:13bb49ad05b0f2d8ebc8d79c79bdfcd299bc7f3eacb959d4552c022807ed5d03
3
+ size 343773880