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