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