TeeA's picture
Training in progress, epoch 0
18bdb02
{
"_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"
}