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