{
  "_name_or_path": "dima806/pokemon_types_image_detection",
  "architectures": [
    "ViTForImageClassification"
  ],
  "attention_probs_dropout_prob": 0.0,
  "encoder_stride": 16,
  "hidden_act": "gelu",
  "hidden_dropout_prob": 0.0,
  "hidden_size": 768,
  "id2label": {
    "0": "Wartortle",
    "1": "Arcanine",
    "2": "Staryu",
    "3": "Arbok",
    "4": "Butterfree",
    "5": "Geodude",
    "6": "Seaking",
    "7": "Diglett",
    "8": "Jynx",
    "9": "Sandslash",
    "10": "Magneton",
    "11": "Scyther",
    "12": "Kabuto",
    "13": "Cubone",
    "14": "Golem",
    "15": "Dewgong",
    "16": "Pidgey",
    "17": "Kadabra",
    "18": "Ditto",
    "19": "Venomoth",
    "20": "Rattata",
    "21": "Alakazam",
    "22": "Machoke",
    "23": "Farfetchd",
    "24": "Omastar",
    "25": "Machamp",
    "26": "Jigglypuff",
    "27": "Dragonite",
    "28": "Weepinbell",
    "29": "Sandshrew",
    "30": "Dugtrio",
    "31": "Mankey",
    "32": "Hitmonchan",
    "33": "Spearow",
    "34": "Caterpie",
    "35": "Dratini",
    "36": "Bulbasaur",
    "37": "Tentacool",
    "38": "Gengar",
    "39": "Machop",
    "40": "Raichu",
    "41": "Alolan Sandslash",
    "42": "Eevee",
    "43": "Abra",
    "44": "Haunter",
    "45": "Metapod",
    "46": "Fearow",
    "47": "Nidorina",
    "48": "Zapdos",
    "49": "Ninetales",
    "50": "Chansey",
    "51": "Kangaskhan",
    "52": "Poliwrath",
    "53": "Gyarados",
    "54": "Charmeleon",
    "55": "Vulpix",
    "56": "Pidgeot",
    "57": "Blastoise",
    "58": "Porygon",
    "59": "Psyduck",
    "60": "Dragonair",
    "61": "Raticate",
    "62": "Squirtle",
    "63": "Charizard",
    "64": "Electrode",
    "65": "Flareon",
    "66": "Exeggutor",
    "67": "Pikachu",
    "68": "Wigglytuff",
    "69": "Venusaur",
    "70": "Mewtwo",
    "71": "Clefable",
    "72": "Oddish",
    "73": "Ekans",
    "74": "Shellder",
    "75": "Marowak",
    "76": "Kakuna",
    "77": "Rapidash",
    "78": "Rhydon",
    "79": "Ivysaur",
    "80": "Slowpoke",
    "81": "Lapras",
    "82": "Clefairy",
    "83": "Hitmonlee",
    "84": "Jolteon",
    "85": "Growlithe",
    "86": "Gastly",
    "87": "Aerodactyl",
    "88": "Weedle",
    "89": "Tauros",
    "90": "Seel",
    "91": "Zubat",
    "92": "Meowth",
    "93": "Persian",
    "94": "Articuno",
    "95": "Weezing",
    "96": "Magnemite",
    "97": "Omanyte",
    "98": "Mew",
    "99": "Vileplume",
    "100": "Nidoqueen",
    "101": "Vaporeon",
    "102": "Ponyta",
    "103": "Moltres",
    "104": "Voltorb",
    "105": "Magikarp",
    "106": "Beedrill",
    "107": "Nidoking",
    "108": "Paras",
    "109": "Grimer",
    "110": "Dodrio",
    "111": "Charmander",
    "112": "Muk",
    "113": "Primeape",
    "114": "Victreebel",
    "115": "Golbat",
    "116": "Horsea",
    "117": "Goldeen",
    "118": "Pidgeotto",
    "119": "Koffing",
    "120": "Seadra",
    "121": "Tentacruel",
    "122": "Pinsir",
    "123": "Cloyster",
    "124": "Gloom",
    "125": "Graveler",
    "126": "Magmar",
    "127": "Krabby",
    "128": "Electabuzz",
    "129": "Poliwhirl",
    "130": "Golduck",
    "131": "Onix",
    "132": "Nidorino",
    "133": "Snorlax",
    "134": "Starmie",
    "135": "Slowbro",
    "136": "MrMime",
    "137": "Venonat",
    "138": "Kabutops",
    "139": "Drowzee",
    "140": "Rhyhorn",
    "141": "Tangela",
    "142": "Doduo",
    "143": "Exeggcute",
    "144": "Poliwag",
    "145": "Lickitung",
    "146": "Hypno",
    "147": "Bellsprout",
    "148": "Parasect",
    "149": "Kingler"
  },
  "image_size": 224,
  "initializer_range": 0.02,
  "intermediate_size": 3072,
  "label2id": {
    "Abra": 43,
    "Aerodactyl": 87,
    "Alakazam": 21,
    "Alolan Sandslash": 41,
    "Arbok": 3,
    "Arcanine": 1,
    "Articuno": 94,
    "Beedrill": 106,
    "Bellsprout": 147,
    "Blastoise": 57,
    "Bulbasaur": 36,
    "Butterfree": 4,
    "Caterpie": 34,
    "Chansey": 50,
    "Charizard": 63,
    "Charmander": 111,
    "Charmeleon": 54,
    "Clefable": 71,
    "Clefairy": 82,
    "Cloyster": 123,
    "Cubone": 13,
    "Dewgong": 15,
    "Diglett": 7,
    "Ditto": 18,
    "Dodrio": 110,
    "Doduo": 142,
    "Dragonair": 60,
    "Dragonite": 27,
    "Dratini": 35,
    "Drowzee": 139,
    "Dugtrio": 30,
    "Eevee": 42,
    "Ekans": 73,
    "Electabuzz": 128,
    "Electrode": 64,
    "Exeggcute": 143,
    "Exeggutor": 66,
    "Farfetchd": 23,
    "Fearow": 46,
    "Flareon": 65,
    "Gastly": 86,
    "Gengar": 38,
    "Geodude": 5,
    "Gloom": 124,
    "Golbat": 115,
    "Goldeen": 117,
    "Golduck": 130,
    "Golem": 14,
    "Graveler": 125,
    "Grimer": 109,
    "Growlithe": 85,
    "Gyarados": 53,
    "Haunter": 44,
    "Hitmonchan": 32,
    "Hitmonlee": 83,
    "Horsea": 116,
    "Hypno": 146,
    "Ivysaur": 79,
    "Jigglypuff": 26,
    "Jolteon": 84,
    "Jynx": 8,
    "Kabuto": 12,
    "Kabutops": 138,
    "Kadabra": 17,
    "Kakuna": 76,
    "Kangaskhan": 51,
    "Kingler": 149,
    "Koffing": 119,
    "Krabby": 127,
    "Lapras": 81,
    "Lickitung": 145,
    "Machamp": 25,
    "Machoke": 22,
    "Machop": 39,
    "Magikarp": 105,
    "Magmar": 126,
    "Magnemite": 96,
    "Magneton": 10,
    "Mankey": 31,
    "Marowak": 75,
    "Meowth": 92,
    "Metapod": 45,
    "Mew": 98,
    "Mewtwo": 70,
    "Moltres": 103,
    "MrMime": 136,
    "Muk": 112,
    "Nidoking": 107,
    "Nidoqueen": 100,
    "Nidorina": 47,
    "Nidorino": 132,
    "Ninetales": 49,
    "Oddish": 72,
    "Omanyte": 97,
    "Omastar": 24,
    "Onix": 131,
    "Paras": 108,
    "Parasect": 148,
    "Persian": 93,
    "Pidgeot": 56,
    "Pidgeotto": 118,
    "Pidgey": 16,
    "Pikachu": 67,
    "Pinsir": 122,
    "Poliwag": 144,
    "Poliwhirl": 129,
    "Poliwrath": 52,
    "Ponyta": 102,
    "Porygon": 58,
    "Primeape": 113,
    "Psyduck": 59,
    "Raichu": 40,
    "Rapidash": 77,
    "Raticate": 61,
    "Rattata": 20,
    "Rhydon": 78,
    "Rhyhorn": 140,
    "Sandshrew": 29,
    "Sandslash": 9,
    "Scyther": 11,
    "Seadra": 120,
    "Seaking": 6,
    "Seel": 90,
    "Shellder": 74,
    "Slowbro": 135,
    "Slowpoke": 80,
    "Snorlax": 133,
    "Spearow": 33,
    "Squirtle": 62,
    "Starmie": 134,
    "Staryu": 2,
    "Tangela": 141,
    "Tauros": 89,
    "Tentacool": 37,
    "Tentacruel": 121,
    "Vaporeon": 101,
    "Venomoth": 19,
    "Venonat": 137,
    "Venusaur": 69,
    "Victreebel": 114,
    "Vileplume": 99,
    "Voltorb": 104,
    "Vulpix": 55,
    "Wartortle": 0,
    "Weedle": 88,
    "Weepinbell": 28,
    "Weezing": 95,
    "Wigglytuff": 68,
    "Zapdos": 48,
    "Zubat": 91
  },
  "layer_norm_eps": 1e-12,
  "model_type": "vit",
  "num_attention_heads": 12,
  "num_channels": 3,
  "num_hidden_layers": 12,
  "patch_size": 16,
  "problem_type": "single_label_classification",
  "qkv_bias": true,
  "torch_dtype": "float32",
  "transformers_version": "4.39.2"
}