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