aesthetics_v3 / config.json
robertsw's picture
Upload Dinov2ForImageClassification
f333271 verified
raw
history blame
4.08 kB
{
"_name_or_path": "facebook/dinov2-giant",
"apply_layernorm": true,
"architectures": [
"Dinov2ForImageClassification"
],
"attention_probs_dropout_prob": 0.0,
"drop_path_rate": 0.0,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 1536,
"id2label": {
"0": "20s",
"1": "30s",
"10": "artsy",
"11": "athleisure",
"12": "balletcore",
"13": "barbiecore",
"14": "beach",
"15": "bimbocore",
"16": "blokette",
"17": "bohemian",
"18": "casual",
"19": "chic",
"2": "40s",
"20": "classicore",
"21": "coastal cowgirl",
"22": "coastal grandmother",
"23": "comfy cozy",
"24": "cool girl",
"25": "coquette",
"26": "corporate",
"27": "country",
"28": "dark academia",
"29": "eclectic",
"3": "50s",
"30": "edgy",
"31": "ethereal",
"32": "europecore",
"33": "folk",
"34": "frazzled english woman",
"35": "fruit girl",
"36": "garconne",
"37": "girly girl",
"38": "goth",
"39": "granola",
"4": "60s",
"40": "grunge",
"41": "indie",
"42": "it girl",
"43": "kawaii",
"44": "lagenlook",
"45": "mermaidcore",
"46": "minimalist",
"47": "mob wife",
"48": "modest",
"49": "nerdy",
"5": "70s",
"50": "normcore",
"51": "old money",
"52": "preppy",
"53": "skater",
"54": "sporty",
"55": "stealth wealth",
"56": "streetwear",
"57": "tomato girl",
"58": "tomboy",
"59": "vanilla girl",
"6": "80s",
"60": "vintage",
"7": "90s",
"8": "Y2K",
"9": "art hoe"
},
"image_size": 518,
"initializer_range": 0.02,
"label2id": {
"20s": "0",
"30s": "1",
"40s": "2",
"50s": "3",
"60s": "4",
"70s": "5",
"80s": "6",
"90s": "7",
"Y2K": "8",
"art hoe": "9",
"artsy": "10",
"athleisure": "11",
"balletcore": "12",
"barbiecore": "13",
"beach": "14",
"bimbocore": "15",
"blokette": "16",
"bohemian": "17",
"casual": "18",
"chic": "19",
"classicore": "20",
"coastal cowgirl": "21",
"coastal grandmother": "22",
"comfy cozy": "23",
"cool girl": "24",
"coquette": "25",
"corporate": "26",
"country": "27",
"dark academia": "28",
"eclectic": "29",
"edgy": "30",
"ethereal": "31",
"europecore": "32",
"folk": "33",
"frazzled english woman": "34",
"fruit girl": "35",
"garconne": "36",
"girly girl": "37",
"goth": "38",
"granola": "39",
"grunge": "40",
"indie": "41",
"it girl": "42",
"kawaii": "43",
"lagenlook": "44",
"mermaidcore": "45",
"minimalist": "46",
"mob wife": "47",
"modest": "48",
"nerdy": "49",
"normcore": "50",
"old money": "51",
"preppy": "52",
"skater": "53",
"sporty": "54",
"stealth wealth": "55",
"streetwear": "56",
"tomato girl": "57",
"tomboy": "58",
"vanilla girl": "59",
"vintage": "60"
},
"layer_norm_eps": 1e-06,
"layerscale_value": 1.0,
"mlp_ratio": 4,
"model_type": "dinov2",
"num_attention_heads": 24,
"num_channels": 3,
"num_hidden_layers": 40,
"out_features": [
"stage40"
],
"out_indices": [
40
],
"patch_size": 14,
"problem_type": "single_label_classification",
"qkv_bias": true,
"reshape_hidden_states": true,
"stage_names": [
"stem",
"stage1",
"stage2",
"stage3",
"stage4",
"stage5",
"stage6",
"stage7",
"stage8",
"stage9",
"stage10",
"stage11",
"stage12",
"stage13",
"stage14",
"stage15",
"stage16",
"stage17",
"stage18",
"stage19",
"stage20",
"stage21",
"stage22",
"stage23",
"stage24",
"stage25",
"stage26",
"stage27",
"stage28",
"stage29",
"stage30",
"stage31",
"stage32",
"stage33",
"stage34",
"stage35",
"stage36",
"stage37",
"stage38",
"stage39",
"stage40"
],
"torch_dtype": "float32",
"transformers_version": "4.38.2",
"use_swiglu_ffn": true
}