| { | |
| "_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" | |
| } | |