agustin228 commited on
Commit
eea32e7
1 Parent(s): 4d2b4c1

Training in progress, epoch 1

Browse files
Files changed (3) hide show
  1. config.json +302 -18
  2. pytorch_model.bin +2 -2
  3. training_args.bin +1 -1
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "google/vit-base-patch16-224-in21k",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
@@ -9,27 +9,311 @@
9
  "hidden_dropout_prob": 0.0,
10
  "hidden_size": 768,
11
  "id2label": {
12
- "0": "anger",
13
- "1": "contempt",
14
- "2": "disgust",
15
- "3": "fear",
16
- "4": "happy",
17
- "5": "neutral",
18
- "6": "sad",
19
- "7": "surprise"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
20
  },
21
  "image_size": 224,
22
  "initializer_range": 0.02,
23
  "intermediate_size": 3072,
24
  "label2id": {
25
- "anger": "0",
26
- "contempt": "1",
27
- "disgust": "2",
28
- "fear": "3",
29
- "happy": "4",
30
- "neutral": "5",
31
- "sad": "6",
32
- "surprise": "7"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
33
  },
34
  "layer_norm_eps": 1e-12,
35
  "model_type": "vit",
@@ -40,5 +324,5 @@
40
  "problem_type": "single_label_classification",
41
  "qkv_bias": true,
42
  "torch_dtype": "float32",
43
- "transformers_version": "4.33.1"
44
  }
 
1
  {
2
+ "_name_or_path": "google/vit-base-patch16-224",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
 
9
  "hidden_dropout_prob": 0.0,
10
  "hidden_size": 768,
11
  "id2label": {
12
+ "0": "Golbat",
13
+ "1": "Machoke",
14
+ "10": "Raichu",
15
+ "100": "Dragonite",
16
+ "101": "Fearow",
17
+ "102": "Slowpoke",
18
+ "103": "Weezing",
19
+ "104": "Beedrill",
20
+ "105": "Weedle",
21
+ "106": "Cloyster",
22
+ "107": "Vaporeon",
23
+ "108": "Gyarados",
24
+ "109": "Golduck",
25
+ "11": "Zapdos",
26
+ "110": "Machamp",
27
+ "111": "Hitmonlee",
28
+ "112": "Primeape",
29
+ "113": "Cubone",
30
+ "114": "Sandslash",
31
+ "115": "Scyther",
32
+ "116": "Haunter",
33
+ "117": "Metapod",
34
+ "118": "Tentacruel",
35
+ "119": "Aerodactyl",
36
+ "12": "Raticate",
37
+ "120": "Kabutops",
38
+ "121": "Ninetales",
39
+ "122": "Zubat",
40
+ "123": "Rhydon",
41
+ "124": "Mew",
42
+ "125": "Pinsir",
43
+ "126": "Ditto",
44
+ "127": "Victreebel",
45
+ "128": "Omanyte",
46
+ "129": "Horsea",
47
+ "13": "Magnemite",
48
+ "130": "Pikachu",
49
+ "131": "Blastoise",
50
+ "132": "Venomoth",
51
+ "133": "Charizard",
52
+ "134": "Seadra",
53
+ "135": "Muk",
54
+ "136": "Spearow",
55
+ "137": "Bulbasaur",
56
+ "138": "Bellsprout",
57
+ "139": "Electrode",
58
+ "14": "Ivysaur",
59
+ "140": "Gloom",
60
+ "141": "Poliwhirl",
61
+ "142": "Flareon",
62
+ "143": "Seaking",
63
+ "144": "Hypno",
64
+ "145": "Wartortle",
65
+ "146": "Mankey",
66
+ "147": "Tentacool",
67
+ "148": "Exeggcute",
68
+ "149": "Meowth",
69
+ "15": "Growlithe",
70
+ "16": "Tangela",
71
+ "17": "Drowzee",
72
+ "18": "Rapidash",
73
+ "19": "Venonat",
74
+ "2": "Omastar",
75
+ "20": "Pidgeot",
76
+ "21": "Nidorino",
77
+ "22": "Porygon",
78
+ "23": "Lickitung",
79
+ "24": "Rattata",
80
+ "25": "Machop",
81
+ "26": "Charmeleon",
82
+ "27": "Slowbro",
83
+ "28": "Parasect",
84
+ "29": "Eevee",
85
+ "3": "Diglett",
86
+ "30": "Starmie",
87
+ "31": "Staryu",
88
+ "32": "Psyduck",
89
+ "33": "Dragonair",
90
+ "34": "Magikarp",
91
+ "35": "Vileplume",
92
+ "36": "Marowak",
93
+ "37": "Pidgeotto",
94
+ "38": "Shellder",
95
+ "39": "Mewtwo",
96
+ "4": "Lapras",
97
+ "40": "Farfetchd",
98
+ "41": "Kingler",
99
+ "42": "Seel",
100
+ "43": "Kakuna",
101
+ "44": "Doduo",
102
+ "45": "Electabuzz",
103
+ "46": "Charmander",
104
+ "47": "Rhyhorn",
105
+ "48": "Tauros",
106
+ "49": "Dugtrio",
107
+ "5": "Kabuto",
108
+ "50": "Poliwrath",
109
+ "51": "Gengar",
110
+ "52": "Exeggutor",
111
+ "53": "Dewgong",
112
+ "54": "Jigglypuff",
113
+ "55": "Geodude",
114
+ "56": "Kadabra",
115
+ "57": "Nidorina",
116
+ "58": "Sandshrew",
117
+ "59": "Grimer",
118
+ "6": "Persian",
119
+ "60": "MrMime",
120
+ "61": "Pidgey",
121
+ "62": "Koffing",
122
+ "63": "Ekans",
123
+ "64": "Alolan Sandslash",
124
+ "65": "Venusaur",
125
+ "66": "Snorlax",
126
+ "67": "Paras",
127
+ "68": "Jynx",
128
+ "69": "Chansey",
129
+ "7": "Weepinbell",
130
+ "70": "Hitmonchan",
131
+ "71": "Gastly",
132
+ "72": "Kangaskhan",
133
+ "73": "Oddish",
134
+ "74": "Wigglytuff",
135
+ "75": "Graveler",
136
+ "76": "Arcanine",
137
+ "77": "Clefairy",
138
+ "78": "Articuno",
139
+ "79": "Poliwag",
140
+ "8": "Golem",
141
+ "80": "Abra",
142
+ "81": "Squirtle",
143
+ "82": "Voltorb",
144
+ "83": "Ponyta",
145
+ "84": "Moltres",
146
+ "85": "Nidoqueen",
147
+ "86": "Magmar",
148
+ "87": "Onix",
149
+ "88": "Vulpix",
150
+ "89": "Butterfree",
151
+ "9": "Dodrio",
152
+ "90": "Krabby",
153
+ "91": "Arbok",
154
+ "92": "Clefable",
155
+ "93": "Goldeen",
156
+ "94": "Magneton",
157
+ "95": "Dratini",
158
+ "96": "Caterpie",
159
+ "97": "Jolteon",
160
+ "98": "Nidoking",
161
+ "99": "Alakazam"
162
  },
163
  "image_size": 224,
164
  "initializer_range": 0.02,
165
  "intermediate_size": 3072,
166
  "label2id": {
167
+ "Abra": "80",
168
+ "Aerodactyl": "119",
169
+ "Alakazam": "99",
170
+ "Alolan Sandslash": "64",
171
+ "Arbok": "91",
172
+ "Arcanine": "76",
173
+ "Articuno": "78",
174
+ "Beedrill": "104",
175
+ "Bellsprout": "138",
176
+ "Blastoise": "131",
177
+ "Bulbasaur": "137",
178
+ "Butterfree": "89",
179
+ "Caterpie": "96",
180
+ "Chansey": "69",
181
+ "Charizard": "133",
182
+ "Charmander": "46",
183
+ "Charmeleon": "26",
184
+ "Clefable": "92",
185
+ "Clefairy": "77",
186
+ "Cloyster": "106",
187
+ "Cubone": "113",
188
+ "Dewgong": "53",
189
+ "Diglett": "3",
190
+ "Ditto": "126",
191
+ "Dodrio": "9",
192
+ "Doduo": "44",
193
+ "Dragonair": "33",
194
+ "Dragonite": "100",
195
+ "Dratini": "95",
196
+ "Drowzee": "17",
197
+ "Dugtrio": "49",
198
+ "Eevee": "29",
199
+ "Ekans": "63",
200
+ "Electabuzz": "45",
201
+ "Electrode": "139",
202
+ "Exeggcute": "148",
203
+ "Exeggutor": "52",
204
+ "Farfetchd": "40",
205
+ "Fearow": "101",
206
+ "Flareon": "142",
207
+ "Gastly": "71",
208
+ "Gengar": "51",
209
+ "Geodude": "55",
210
+ "Gloom": "140",
211
+ "Golbat": "0",
212
+ "Goldeen": "93",
213
+ "Golduck": "109",
214
+ "Golem": "8",
215
+ "Graveler": "75",
216
+ "Grimer": "59",
217
+ "Growlithe": "15",
218
+ "Gyarados": "108",
219
+ "Haunter": "116",
220
+ "Hitmonchan": "70",
221
+ "Hitmonlee": "111",
222
+ "Horsea": "129",
223
+ "Hypno": "144",
224
+ "Ivysaur": "14",
225
+ "Jigglypuff": "54",
226
+ "Jolteon": "97",
227
+ "Jynx": "68",
228
+ "Kabuto": "5",
229
+ "Kabutops": "120",
230
+ "Kadabra": "56",
231
+ "Kakuna": "43",
232
+ "Kangaskhan": "72",
233
+ "Kingler": "41",
234
+ "Koffing": "62",
235
+ "Krabby": "90",
236
+ "Lapras": "4",
237
+ "Lickitung": "23",
238
+ "Machamp": "110",
239
+ "Machoke": "1",
240
+ "Machop": "25",
241
+ "Magikarp": "34",
242
+ "Magmar": "86",
243
+ "Magnemite": "13",
244
+ "Magneton": "94",
245
+ "Mankey": "146",
246
+ "Marowak": "36",
247
+ "Meowth": "149",
248
+ "Metapod": "117",
249
+ "Mew": "124",
250
+ "Mewtwo": "39",
251
+ "Moltres": "84",
252
+ "MrMime": "60",
253
+ "Muk": "135",
254
+ "Nidoking": "98",
255
+ "Nidoqueen": "85",
256
+ "Nidorina": "57",
257
+ "Nidorino": "21",
258
+ "Ninetales": "121",
259
+ "Oddish": "73",
260
+ "Omanyte": "128",
261
+ "Omastar": "2",
262
+ "Onix": "87",
263
+ "Paras": "67",
264
+ "Parasect": "28",
265
+ "Persian": "6",
266
+ "Pidgeot": "20",
267
+ "Pidgeotto": "37",
268
+ "Pidgey": "61",
269
+ "Pikachu": "130",
270
+ "Pinsir": "125",
271
+ "Poliwag": "79",
272
+ "Poliwhirl": "141",
273
+ "Poliwrath": "50",
274
+ "Ponyta": "83",
275
+ "Porygon": "22",
276
+ "Primeape": "112",
277
+ "Psyduck": "32",
278
+ "Raichu": "10",
279
+ "Rapidash": "18",
280
+ "Raticate": "12",
281
+ "Rattata": "24",
282
+ "Rhydon": "123",
283
+ "Rhyhorn": "47",
284
+ "Sandshrew": "58",
285
+ "Sandslash": "114",
286
+ "Scyther": "115",
287
+ "Seadra": "134",
288
+ "Seaking": "143",
289
+ "Seel": "42",
290
+ "Shellder": "38",
291
+ "Slowbro": "27",
292
+ "Slowpoke": "102",
293
+ "Snorlax": "66",
294
+ "Spearow": "136",
295
+ "Squirtle": "81",
296
+ "Starmie": "30",
297
+ "Staryu": "31",
298
+ "Tangela": "16",
299
+ "Tauros": "48",
300
+ "Tentacool": "147",
301
+ "Tentacruel": "118",
302
+ "Vaporeon": "107",
303
+ "Venomoth": "132",
304
+ "Venonat": "19",
305
+ "Venusaur": "65",
306
+ "Victreebel": "127",
307
+ "Vileplume": "35",
308
+ "Voltorb": "82",
309
+ "Vulpix": "88",
310
+ "Wartortle": "145",
311
+ "Weedle": "105",
312
+ "Weepinbell": "7",
313
+ "Weezing": "103",
314
+ "Wigglytuff": "74",
315
+ "Zapdos": "11",
316
+ "Zubat": "122"
317
  },
318
  "layer_norm_eps": 1e-12,
319
  "model_type": "vit",
 
324
  "problem_type": "single_label_classification",
325
  "qkv_bias": true,
326
  "torch_dtype": "float32",
327
+ "transformers_version": "4.33.3"
328
  }
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:cbdfd141f27de8cccc37a5fb29fcf111d013098ce703c2838040cb2768466cd0
3
- size 343287149
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9873e27516f12d5bc48cfe3abbb9d49c50065c193210815fd5d8fa5f9beb68c9
3
+ size 343723949
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b4ea8e6fa3bbc29cbc67861635228737044805d357d40d96e6d7a450bd7f06c4
3
  size 4091
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:148aeea34bf3f1de98ae55f80b4e4541f9e640c72c5f58d5d91961665233051b
3
  size 4091