TuALe commited on
Commit
ba935fa
1 Parent(s): 9e17705

End of training

Browse files
Files changed (4) hide show
  1. README.md +58 -0
  2. config.json +326 -0
  3. preprocessor_config.json +22 -0
  4. tf_model.h5 +3 -0
README.md ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - generated_from_keras_callback
5
+ model-index:
6
+ - name: TuALe/Tu_Le_Pokemon_Classifier
7
+ results: []
8
+ ---
9
+
10
+ <!-- This model card has been generated automatically according to the information Keras had access to. You should
11
+ probably proofread and complete it, then remove this comment. -->
12
+
13
+ # TuALe/Tu_Le_Pokemon_Classifier
14
+
15
+ 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.
16
+ It achieves the following results on the evaluation set:
17
+ - Train Loss: 2.6112
18
+ - Validation Loss: 2.5356
19
+ - Train Accuracy: 0.915
20
+ - Epoch: 4
21
+
22
+ ## Model description
23
+
24
+ More information needed
25
+
26
+ ## Intended uses & limitations
27
+
28
+ More information needed
29
+
30
+ ## Training and evaluation data
31
+
32
+ More information needed
33
+
34
+ ## Training procedure
35
+
36
+ ### Training hyperparameters
37
+
38
+ The following hyperparameters were used during training:
39
+ - optimizer: {'name': 'AdamWeightDecay', 'learning_rate': {'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 3e-05, 'decay_steps': 4000, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}}, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False, 'weight_decay_rate': 0.01}
40
+ - training_precision: float32
41
+
42
+ ### Training results
43
+
44
+ | Train Loss | Validation Loss | Train Accuracy | Epoch |
45
+ |:----------:|:---------------:|:--------------:|:-----:|
46
+ | 4.6606 | 4.3303 | 0.475 | 0 |
47
+ | 4.0521 | 3.8109 | 0.78 | 1 |
48
+ | 3.5169 | 3.3200 | 0.875 | 2 |
49
+ | 3.0280 | 2.8918 | 0.885 | 3 |
50
+ | 2.6112 | 2.5356 | 0.915 | 4 |
51
+
52
+
53
+ ### Framework versions
54
+
55
+ - Transformers 4.29.2
56
+ - TensorFlow 2.12.0
57
+ - Datasets 2.6.1
58
+ - Tokenizers 0.13.3
config.json ADDED
@@ -0,0 +1,326 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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": "Porygon",
13
+ "1": "Goldeen",
14
+ "2": "Hitmonlee",
15
+ "3": "Hitmonchan",
16
+ "4": "Gloom",
17
+ "5": "Aerodactyl",
18
+ "6": "Mankey",
19
+ "7": "Seadra",
20
+ "8": "Gengar",
21
+ "9": "Venonat",
22
+ "10": "Articuno",
23
+ "11": "Seaking",
24
+ "12": "Dugtrio",
25
+ "13": "Machop",
26
+ "14": "Jynx",
27
+ "15": "Oddish",
28
+ "16": "Dodrio",
29
+ "17": "Dragonair",
30
+ "18": "Weedle",
31
+ "19": "Golduck",
32
+ "20": "Flareon",
33
+ "21": "Krabby",
34
+ "22": "Parasect",
35
+ "23": "Ninetales",
36
+ "24": "Nidoqueen",
37
+ "25": "Kabutops",
38
+ "26": "Drowzee",
39
+ "27": "Caterpie",
40
+ "28": "Jigglypuff",
41
+ "29": "Machamp",
42
+ "30": "Clefairy",
43
+ "31": "Kangaskhan",
44
+ "32": "Dragonite",
45
+ "33": "Weepinbell",
46
+ "34": "Fearow",
47
+ "35": "Bellsprout",
48
+ "36": "Grimer",
49
+ "37": "Nidorina",
50
+ "38": "Staryu",
51
+ "39": "Horsea",
52
+ "40": "Electabuzz",
53
+ "41": "Dratini",
54
+ "42": "Machoke",
55
+ "43": "Magnemite",
56
+ "44": "Squirtle",
57
+ "45": "Gyarados",
58
+ "46": "Pidgeot",
59
+ "47": "Bulbasaur",
60
+ "48": "Nidoking",
61
+ "49": "Golem",
62
+ "50": "Dewgong",
63
+ "51": "Moltres",
64
+ "52": "Zapdos",
65
+ "53": "Poliwrath",
66
+ "54": "Vulpix",
67
+ "55": "Beedrill",
68
+ "56": "Charmander",
69
+ "57": "Abra",
70
+ "58": "Zubat",
71
+ "59": "Golbat",
72
+ "60": "Wigglytuff",
73
+ "61": "Charizard",
74
+ "62": "Slowpoke",
75
+ "63": "Poliwag",
76
+ "64": "Tentacruel",
77
+ "65": "Rhyhorn",
78
+ "66": "Onix",
79
+ "67": "Butterfree",
80
+ "68": "Exeggcute",
81
+ "69": "Sandslash",
82
+ "70": "Pinsir",
83
+ "71": "Rattata",
84
+ "72": "Growlithe",
85
+ "73": "Haunter",
86
+ "74": "Pidgey",
87
+ "75": "Ditto",
88
+ "76": "Farfetchd",
89
+ "77": "Pikachu",
90
+ "78": "Raticate",
91
+ "79": "Wartortle",
92
+ "80": "Vaporeon",
93
+ "81": "Cloyster",
94
+ "82": "Hypno",
95
+ "83": "Arbok",
96
+ "84": "Metapod",
97
+ "85": "Tangela",
98
+ "86": "Kingler",
99
+ "87": "Exeggutor",
100
+ "88": "Kadabra",
101
+ "89": "Seel",
102
+ "90": "Voltorb",
103
+ "91": "Chansey",
104
+ "92": "Venomoth",
105
+ "93": "Ponyta",
106
+ "94": "Vileplume",
107
+ "95": "Koffing",
108
+ "96": "Blastoise",
109
+ "97": "Tentacool",
110
+ "98": "Lickitung",
111
+ "99": "Paras",
112
+ "100": "Clefable",
113
+ "101": "Cubone",
114
+ "102": "Marowak",
115
+ "103": "Nidorino",
116
+ "104": "Jolteon",
117
+ "105": "Muk",
118
+ "106": "Magikarp",
119
+ "107": "Slowbro",
120
+ "108": "Tauros",
121
+ "109": "Kabuto",
122
+ "110": "Spearow",
123
+ "111": "Sandshrew",
124
+ "112": "Eevee",
125
+ "113": "Kakuna",
126
+ "114": "Omastar",
127
+ "115": "Ekans",
128
+ "116": "Geodude",
129
+ "117": "Magmar",
130
+ "118": "Snorlax",
131
+ "119": "Meowth",
132
+ "120": "Pidgeotto",
133
+ "121": "Venusaur",
134
+ "122": "Persian",
135
+ "123": "Rhydon",
136
+ "124": "Starmie",
137
+ "125": "Charmeleon",
138
+ "126": "Lapras",
139
+ "127": "Alakazam",
140
+ "128": "Graveler",
141
+ "129": "Psyduck",
142
+ "130": "Rapidash",
143
+ "131": "Doduo",
144
+ "132": "Magneton",
145
+ "133": "Arcanine",
146
+ "134": "Electrode",
147
+ "135": "Omanyte",
148
+ "136": "Poliwhirl",
149
+ "137": "Mew",
150
+ "138": "Alolan Sandslash",
151
+ "139": "Mewtwo",
152
+ "140": "Weezing",
153
+ "141": "Gastly",
154
+ "142": "Victreebel",
155
+ "143": "Ivysaur",
156
+ "144": "MrMime",
157
+ "145": "Shellder",
158
+ "146": "Scyther",
159
+ "147": "Diglett",
160
+ "148": "Primeape",
161
+ "149": "Raichu"
162
+ },
163
+ "image_size": 224,
164
+ "initializer_range": 0.02,
165
+ "intermediate_size": 3072,
166
+ "label2id": {
167
+ "Abra": 57,
168
+ "Aerodactyl": 5,
169
+ "Alakazam": 127,
170
+ "Alolan Sandslash": 138,
171
+ "Arbok": 83,
172
+ "Arcanine": 133,
173
+ "Articuno": 10,
174
+ "Beedrill": 55,
175
+ "Bellsprout": 35,
176
+ "Blastoise": 96,
177
+ "Bulbasaur": 47,
178
+ "Butterfree": 67,
179
+ "Caterpie": 27,
180
+ "Chansey": 91,
181
+ "Charizard": 61,
182
+ "Charmander": 56,
183
+ "Charmeleon": 125,
184
+ "Clefable": 100,
185
+ "Clefairy": 30,
186
+ "Cloyster": 81,
187
+ "Cubone": 101,
188
+ "Dewgong": 50,
189
+ "Diglett": 147,
190
+ "Ditto": 75,
191
+ "Dodrio": 16,
192
+ "Doduo": 131,
193
+ "Dragonair": 17,
194
+ "Dragonite": 32,
195
+ "Dratini": 41,
196
+ "Drowzee": 26,
197
+ "Dugtrio": 12,
198
+ "Eevee": 112,
199
+ "Ekans": 115,
200
+ "Electabuzz": 40,
201
+ "Electrode": 134,
202
+ "Exeggcute": 68,
203
+ "Exeggutor": 87,
204
+ "Farfetchd": 76,
205
+ "Fearow": 34,
206
+ "Flareon": 20,
207
+ "Gastly": 141,
208
+ "Gengar": 8,
209
+ "Geodude": 116,
210
+ "Gloom": 4,
211
+ "Golbat": 59,
212
+ "Goldeen": 1,
213
+ "Golduck": 19,
214
+ "Golem": 49,
215
+ "Graveler": 128,
216
+ "Grimer": 36,
217
+ "Growlithe": 72,
218
+ "Gyarados": 45,
219
+ "Haunter": 73,
220
+ "Hitmonchan": 3,
221
+ "Hitmonlee": 2,
222
+ "Horsea": 39,
223
+ "Hypno": 82,
224
+ "Ivysaur": 143,
225
+ "Jigglypuff": 28,
226
+ "Jolteon": 104,
227
+ "Jynx": 14,
228
+ "Kabuto": 109,
229
+ "Kabutops": 25,
230
+ "Kadabra": 88,
231
+ "Kakuna": 113,
232
+ "Kangaskhan": 31,
233
+ "Kingler": 86,
234
+ "Koffing": 95,
235
+ "Krabby": 21,
236
+ "Lapras": 126,
237
+ "Lickitung": 98,
238
+ "Machamp": 29,
239
+ "Machoke": 42,
240
+ "Machop": 13,
241
+ "Magikarp": 106,
242
+ "Magmar": 117,
243
+ "Magnemite": 43,
244
+ "Magneton": 132,
245
+ "Mankey": 6,
246
+ "Marowak": 102,
247
+ "Meowth": 119,
248
+ "Metapod": 84,
249
+ "Mew": 137,
250
+ "Mewtwo": 139,
251
+ "Moltres": 51,
252
+ "MrMime": 144,
253
+ "Muk": 105,
254
+ "Nidoking": 48,
255
+ "Nidoqueen": 24,
256
+ "Nidorina": 37,
257
+ "Nidorino": 103,
258
+ "Ninetales": 23,
259
+ "Oddish": 15,
260
+ "Omanyte": 135,
261
+ "Omastar": 114,
262
+ "Onix": 66,
263
+ "Paras": 99,
264
+ "Parasect": 22,
265
+ "Persian": 122,
266
+ "Pidgeot": 46,
267
+ "Pidgeotto": 120,
268
+ "Pidgey": 74,
269
+ "Pikachu": 77,
270
+ "Pinsir": 70,
271
+ "Poliwag": 63,
272
+ "Poliwhirl": 136,
273
+ "Poliwrath": 53,
274
+ "Ponyta": 93,
275
+ "Porygon": 0,
276
+ "Primeape": 148,
277
+ "Psyduck": 129,
278
+ "Raichu": 149,
279
+ "Rapidash": 130,
280
+ "Raticate": 78,
281
+ "Rattata": 71,
282
+ "Rhydon": 123,
283
+ "Rhyhorn": 65,
284
+ "Sandshrew": 111,
285
+ "Sandslash": 69,
286
+ "Scyther": 146,
287
+ "Seadra": 7,
288
+ "Seaking": 11,
289
+ "Seel": 89,
290
+ "Shellder": 145,
291
+ "Slowbro": 107,
292
+ "Slowpoke": 62,
293
+ "Snorlax": 118,
294
+ "Spearow": 110,
295
+ "Squirtle": 44,
296
+ "Starmie": 124,
297
+ "Staryu": 38,
298
+ "Tangela": 85,
299
+ "Tauros": 108,
300
+ "Tentacool": 97,
301
+ "Tentacruel": 64,
302
+ "Vaporeon": 80,
303
+ "Venomoth": 92,
304
+ "Venonat": 9,
305
+ "Venusaur": 121,
306
+ "Victreebel": 142,
307
+ "Vileplume": 94,
308
+ "Voltorb": 90,
309
+ "Vulpix": 54,
310
+ "Wartortle": 79,
311
+ "Weedle": 18,
312
+ "Weepinbell": 33,
313
+ "Weezing": 140,
314
+ "Wigglytuff": 60,
315
+ "Zapdos": 52,
316
+ "Zubat": 58
317
+ },
318
+ "layer_norm_eps": 1e-12,
319
+ "model_type": "vit",
320
+ "num_attention_heads": 12,
321
+ "num_channels": 3,
322
+ "num_hidden_layers": 12,
323
+ "patch_size": 16,
324
+ "qkv_bias": true,
325
+ "transformers_version": "4.29.2"
326
+ }
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:8f9b766815678f1060a1551f8f91fae13b2fed9cdda936a8ac8804b9d5e3f0af
3
+ size 343924408