minatosnow's picture
Training in progress, epoch 0
cba851e
{
"_name_or_path": "microsoft/swinv2-small-patch4-window16-256",
"architectures": [
"Swinv2ForImageClassification"
],
"attention_probs_dropout_prob": 0.0,
"depths": [
2,
2,
18,
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": "acanthite",
"1": "actinolite",
"2": "adamite",
"3": "aegirine",
"4": "afghanite",
"5": "agate",
"6": "alabandite",
"7": "albite",
"8": "almandine",
"9": "amethyst",
"10": "analcime",
"11": "anatase",
"12": "andalusite",
"13": "andradite",
"14": "anglesite",
"15": "anhydrite",
"16": "annabergite",
"17": "anorthite",
"18": "anorthoclase",
"19": "antimony",
"20": "apatite",
"21": "apophyllite",
"22": "aquamarine",
"23": "aragonite",
"24": "arfvedsonite",
"25": "arsenic",
"26": "arsenopyrite",
"27": "augite",
"28": "aurichalcite",
"29": "axinite",
"30": "azurite",
"31": "babingtonite",
"32": "barite",
"33": "bauxite",
"34": "benitoite",
"35": "beryl",
"36": "biotite",
"37": "bismuth",
"38": "bismuthinite",
"39": "bixbyite",
"40": "boehmite",
"41": "boleite",
"42": "boracite",
"43": "borax",
"44": "bornite",
"45": "boulangerite",
"46": "bournonite",
"47": "brochantite",
"48": "bromargyrite",
"49": "brookite",
"50": "brucite",
"51": "buergerite",
"52": "bustamite",
"53": "bytownite",
"54": "calaverite",
"55": "calcite",
"56": "calomel",
"57": "carletonite",
"58": "cassiterite",
"59": "celestine",
"60": "cerussite",
"61": "cervantite",
"62": "chabazite",
"63": "chalcanthite",
"64": "chalcedony",
"65": "chalcocite",
"66": "chalcopyrite",
"67": "chambersite",
"68": "chamosite",
"69": "chlorargyrite",
"70": "chlorite",
"71": "chondrodite",
"72": "chromite",
"73": "chrysoberyl",
"74": "chrysocolla",
"75": "cinnabar",
"76": "citrine",
"77": "clinochlore",
"78": "clinohumite",
"79": "clinozoisite",
"80": "cobaltite",
"81": "coesite",
"82": "colemanite",
"83": "columbite",
"84": "cookeite",
"85": "copper",
"86": "cordierite",
"87": "corundum",
"88": "covellite",
"89": "cristobalite",
"90": "crocoite",
"91": "cryolite",
"92": "cumengeite",
"93": "cuprite",
"94": "cyanotrichite",
"95": "danburite",
"96": "datolite",
"97": "diamond",
"98": "diaspore",
"99": "diopside",
"100": "dioptase",
"101": "dolomite",
"102": "dravite",
"103": "dumortierite",
"104": "edenite",
"105": "elbaite",
"106": "emerald",
"107": "enargite",
"108": "enstatite",
"109": "epidote",
"110": "epistilbite",
"111": "erythrite",
"112": "euclase",
"113": "ferberite",
"114": "fluorite",
"115": "franklinite",
"116": "galena",
"117": "garnet",
"118": "gaspeite",
"119": "gibbsite",
"120": "glauberite",
"121": "glaucophane",
"122": "gmelinite",
"123": "goethite",
"124": "gold",
"125": "goosecreekite",
"126": "graphite",
"127": "greenockite",
"128": "grossular",
"129": "gummite",
"130": "gypsum",
"131": "halite",
"132": "hastingsite",
"133": "hauyne",
"134": "hedenbergite",
"135": "hematite",
"136": "hemimorphite",
"137": "heulandite",
"138": "hornblende",
"139": "howlite",
"140": "huebnerite",
"141": "humite",
"142": "hydrozincite",
"143": "ice",
"144": "ilmenite",
"145": "inesite",
"146": "iodargyrite",
"147": "iron-nickel",
"148": "jadeite",
"149": "jamesonite",
"150": "jarosite",
"151": "jasper",
"152": "jeremejevite",
"153": "johannsenite",
"154": "kaolinite",
"155": "kernite",
"156": "krennerite",
"157": "kyanite",
"158": "labradorite",
"159": "laumontite",
"160": "lawsonite",
"161": "lazulite",
"162": "lazurite",
"163": "lead",
"164": "lepidolite",
"165": "leucite",
"166": "liddicoatite",
"167": "limonite",
"168": "linarite",
"169": "lithiophilite",
"170": "loellingite",
"171": "luzonite",
"172": "magnesiochromite",
"173": "magnesite",
"174": "magnetite",
"175": "malachite",
"176": "manganite",
"177": "manganoneptunite",
"178": "marcasite",
"179": "marialite",
"180": "meionite",
"181": "melanophlogite",
"182": "mercury",
"183": "mesolite",
"184": "metacinnabar",
"185": "metastibnite",
"186": "metavariscite",
"187": "miargyrite",
"188": "microcline",
"189": "millerite",
"190": "mimetite",
"191": "molybdenite",
"192": "monazite",
"193": "muscovite",
"194": "natrolite",
"195": "nepheline",
"196": "neptunite",
"197": "nickeline",
"198": "nitratine",
"199": "norbergite",
"200": "nosean",
"201": "oligoclase",
"202": "olivine",
"203": "olmiite",
"204": "opal",
"205": "orpiment",
"206": "orthoclase",
"207": "otavite",
"208": "paradamite",
"209": "pararealgar",
"210": "pargasite",
"211": "pearceite",
"212": "pectolite",
"213": "pentlandite",
"214": "pezzottaite",
"215": "phenakite",
"216": "phlogopite",
"217": "phosgenite",
"218": "plagioclase",
"219": "platinum",
"220": "polybasite",
"221": "powellite",
"222": "prehnite",
"223": "proustite",
"224": "psilomelane",
"225": "pumpellyite",
"226": "pyrargyrite",
"227": "pyrite",
"228": "pyrolusite",
"229": "pyromorphite",
"230": "pyrope",
"231": "pyrophyllite",
"232": "pyroxmangite",
"233": "pyrrhotite",
"234": "quartz",
"235": "rammelsbergite",
"236": "raspite",
"237": "realgar",
"238": "rheniite",
"239": "rhodochrosite",
"240": "rhodonite",
"241": "riebeckite",
"242": "romanechite",
"243": "rutile",
"244": "talc",
"245": "tantalite",
"246": "tellurium",
"247": "tennantite",
"248": "tephroite",
"249": "tetrahedrite",
"250": "thenardite",
"251": "thomsonite",
"252": "thorite",
"253": "tincalconite",
"254": "titanite",
"255": "topaz",
"256": "tourmaline",
"257": "tremolite",
"258": "tridymite",
"259": "triphylite",
"260": "turquoise",
"261": "ulexite",
"262": "uraninite",
"263": "uvarovite",
"264": "uvite",
"265": "vanadinite",
"266": "variscite",
"267": "vesuvianite",
"268": "vivianite",
"269": "water",
"270": "wavellite",
"271": "willemite",
"272": "witherite",
"273": "wolframite",
"274": "wollastonite",
"275": "wulfenite",
"276": "wurtzite",
"277": "xanthoconite",
"278": "xenotime",
"279": "zincite",
"280": "zircon",
"281": "zoisite"
},
"image_size": 256,
"initializer_range": 0.02,
"label2id": {
"acanthite": 0,
"actinolite": 1,
"adamite": 2,
"aegirine": 3,
"afghanite": 4,
"agate": 5,
"alabandite": 6,
"albite": 7,
"almandine": 8,
"amethyst": 9,
"analcime": 10,
"anatase": 11,
"andalusite": 12,
"andradite": 13,
"anglesite": 14,
"anhydrite": 15,
"annabergite": 16,
"anorthite": 17,
"anorthoclase": 18,
"antimony": 19,
"apatite": 20,
"apophyllite": 21,
"aquamarine": 22,
"aragonite": 23,
"arfvedsonite": 24,
"arsenic": 25,
"arsenopyrite": 26,
"augite": 27,
"aurichalcite": 28,
"axinite": 29,
"azurite": 30,
"babingtonite": 31,
"barite": 32,
"bauxite": 33,
"benitoite": 34,
"beryl": 35,
"biotite": 36,
"bismuth": 37,
"bismuthinite": 38,
"bixbyite": 39,
"boehmite": 40,
"boleite": 41,
"boracite": 42,
"borax": 43,
"bornite": 44,
"boulangerite": 45,
"bournonite": 46,
"brochantite": 47,
"bromargyrite": 48,
"brookite": 49,
"brucite": 50,
"buergerite": 51,
"bustamite": 52,
"bytownite": 53,
"calaverite": 54,
"calcite": 55,
"calomel": 56,
"carletonite": 57,
"cassiterite": 58,
"celestine": 59,
"cerussite": 60,
"cervantite": 61,
"chabazite": 62,
"chalcanthite": 63,
"chalcedony": 64,
"chalcocite": 65,
"chalcopyrite": 66,
"chambersite": 67,
"chamosite": 68,
"chlorargyrite": 69,
"chlorite": 70,
"chondrodite": 71,
"chromite": 72,
"chrysoberyl": 73,
"chrysocolla": 74,
"cinnabar": 75,
"citrine": 76,
"clinochlore": 77,
"clinohumite": 78,
"clinozoisite": 79,
"cobaltite": 80,
"coesite": 81,
"colemanite": 82,
"columbite": 83,
"cookeite": 84,
"copper": 85,
"cordierite": 86,
"corundum": 87,
"covellite": 88,
"cristobalite": 89,
"crocoite": 90,
"cryolite": 91,
"cumengeite": 92,
"cuprite": 93,
"cyanotrichite": 94,
"danburite": 95,
"datolite": 96,
"diamond": 97,
"diaspore": 98,
"diopside": 99,
"dioptase": 100,
"dolomite": 101,
"dravite": 102,
"dumortierite": 103,
"edenite": 104,
"elbaite": 105,
"emerald": 106,
"enargite": 107,
"enstatite": 108,
"epidote": 109,
"epistilbite": 110,
"erythrite": 111,
"euclase": 112,
"ferberite": 113,
"fluorite": 114,
"franklinite": 115,
"galena": 116,
"garnet": 117,
"gaspeite": 118,
"gibbsite": 119,
"glauberite": 120,
"glaucophane": 121,
"gmelinite": 122,
"goethite": 123,
"gold": 124,
"goosecreekite": 125,
"graphite": 126,
"greenockite": 127,
"grossular": 128,
"gummite": 129,
"gypsum": 130,
"halite": 131,
"hastingsite": 132,
"hauyne": 133,
"hedenbergite": 134,
"hematite": 135,
"hemimorphite": 136,
"heulandite": 137,
"hornblende": 138,
"howlite": 139,
"huebnerite": 140,
"humite": 141,
"hydrozincite": 142,
"ice": 143,
"ilmenite": 144,
"inesite": 145,
"iodargyrite": 146,
"iron-nickel": 147,
"jadeite": 148,
"jamesonite": 149,
"jarosite": 150,
"jasper": 151,
"jeremejevite": 152,
"johannsenite": 153,
"kaolinite": 154,
"kernite": 155,
"krennerite": 156,
"kyanite": 157,
"labradorite": 158,
"laumontite": 159,
"lawsonite": 160,
"lazulite": 161,
"lazurite": 162,
"lead": 163,
"lepidolite": 164,
"leucite": 165,
"liddicoatite": 166,
"limonite": 167,
"linarite": 168,
"lithiophilite": 169,
"loellingite": 170,
"luzonite": 171,
"magnesiochromite": 172,
"magnesite": 173,
"magnetite": 174,
"malachite": 175,
"manganite": 176,
"manganoneptunite": 177,
"marcasite": 178,
"marialite": 179,
"meionite": 180,
"melanophlogite": 181,
"mercury": 182,
"mesolite": 183,
"metacinnabar": 184,
"metastibnite": 185,
"metavariscite": 186,
"miargyrite": 187,
"microcline": 188,
"millerite": 189,
"mimetite": 190,
"molybdenite": 191,
"monazite": 192,
"muscovite": 193,
"natrolite": 194,
"nepheline": 195,
"neptunite": 196,
"nickeline": 197,
"nitratine": 198,
"norbergite": 199,
"nosean": 200,
"oligoclase": 201,
"olivine": 202,
"olmiite": 203,
"opal": 204,
"orpiment": 205,
"orthoclase": 206,
"otavite": 207,
"paradamite": 208,
"pararealgar": 209,
"pargasite": 210,
"pearceite": 211,
"pectolite": 212,
"pentlandite": 213,
"pezzottaite": 214,
"phenakite": 215,
"phlogopite": 216,
"phosgenite": 217,
"plagioclase": 218,
"platinum": 219,
"polybasite": 220,
"powellite": 221,
"prehnite": 222,
"proustite": 223,
"psilomelane": 224,
"pumpellyite": 225,
"pyrargyrite": 226,
"pyrite": 227,
"pyrolusite": 228,
"pyromorphite": 229,
"pyrope": 230,
"pyrophyllite": 231,
"pyroxmangite": 232,
"pyrrhotite": 233,
"quartz": 234,
"rammelsbergite": 235,
"raspite": 236,
"realgar": 237,
"rheniite": 238,
"rhodochrosite": 239,
"rhodonite": 240,
"riebeckite": 241,
"romanechite": 242,
"rutile": 243,
"talc": 244,
"tantalite": 245,
"tellurium": 246,
"tennantite": 247,
"tephroite": 248,
"tetrahedrite": 249,
"thenardite": 250,
"thomsonite": 251,
"thorite": 252,
"tincalconite": 253,
"titanite": 254,
"topaz": 255,
"tourmaline": 256,
"tremolite": 257,
"tridymite": 258,
"triphylite": 259,
"turquoise": 260,
"ulexite": 261,
"uraninite": 262,
"uvarovite": 263,
"uvite": 264,
"vanadinite": 265,
"variscite": 266,
"vesuvianite": 267,
"vivianite": 268,
"water": 269,
"wavellite": 270,
"willemite": 271,
"witherite": 272,
"wolframite": 273,
"wollastonite": 274,
"wulfenite": 275,
"wurtzite": 276,
"xanthoconite": 277,
"xenotime": 278,
"zincite": 279,
"zircon": 280,
"zoisite": 281
},
"layer_norm_eps": 1e-05,
"mlp_ratio": 4.0,
"model_type": "swinv2",
"num_channels": 3,
"num_heads": [
3,
6,
12,
24
],
"num_layers": 4,
"patch_size": 4,
"path_norm": true,
"pretrained_window_sizes": [
0,
0,
0,
0
],
"problem_type": "single_label_classification",
"qkv_bias": true,
"torch_dtype": "float32",
"transformers_version": "4.31.0",
"use_absolute_embeddings": false,
"window_size": 16
}