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