IAmFlyingMonkey
/

pokemon_classifier

Image Classification

Model card Files Files and versions

IAmFlyingMonkey commited on May 27, 2024

Commit

1546f5f

·

verified ·

1 Parent(s): fd9a10b

Update config.json

Files changed (1) hide show

config.json +40 -40

config.json CHANGED Viewed

@@ -9,51 +9,51 @@
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
-    "0": "Bulbasaur",
-    "1": "Ivysaur",
-    "2": "Venusaur",
-    "3": "Charmander",
-    "4": "Charmeleon",
-    "5": "Charizard",
-    "6": "Squirtle",
-    "7": "Wartortle",
-    "8": "Blastoise",
-    "9": "Caterpie",
-    "10": "Metapod",
-    "11": "Butterfree",
-    "12": "Weedle",
-    "13": "Kakuna",
-    "14": "Beedrill",
-    "15": "Pidgey",
-    "16": "Pidgeotto",
-    "17": "Pidgeot",
-    "18": "Rattata",
-    "19": "Raticate"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "Bulbasaur": "0",
-    "Ivysaur": "1",
-    "Venusaur": "2",
-    "Charmander": "3",
-    "Charmeleon": "4",
-    "Charizard": "5",
-    "Squirtle": "6",
-    "Wartortle": "7",
-    "Blastoise": "8",
-    "Caterpie": "9",
-    "Metapod": "10",
-    "Butterfree": "11",
-    "Weedle": "12",
-    "Kakuna": "13",
-    "Beedrill": "14",
-    "Pidgey": "15",
-    "Pidgeotto": "16",
-    "Pidgeot": "17",
-    "Rattata": "18",
-    "Raticate": "19"
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",

   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
+    "0": "Coterpie",
+    "1": "Chikorita",
+    "2": "Cyndaquil",
+    "3": "Ekans",
+    "4": "Geodude",
+    "5": "Hoothoot",
+    "6": "Jigglypuff",
+    "7": "Machop",
+    "8": "Magikarp",
+    "9": "Mareep",
+    "10": "Oddish",
+    "11": "Pidgey",
+    "12": "Pikachu",
+    "13": "Rattata",
+    "14": "Sudowoodo",
+    "15": "Totodile",
+    "16": "Tyrogue",
+    "17": "Vulpix",
+    "18": "Weedle",
+    "19": "Zubat"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
+    "Coterpie": "0",
+    "Chikorita": "1",
+    "Cyndaquil": "2",
+    "Ekans": "3",
+    "Geodude": "4",
+    "Hoothoot": "5",
+    "Jigglypuff": "6",
+    "Machop": "7",
+    "Magikarp": "8",
+    "Mareep": "9",
+    "Oddish": "10",
+    "Pidgey": "11",
+    "Pikachu": "12",
+    "Rattata": "13",
+    "Sudowoodo": "14",
+    "Totodile": "15",
+    "Tyrogue": "16",
+    "Vulpix": "17",
+    "Weedle": "18",
+    "Zubat": "19"
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",