dino_base_H100 / config.json
robertsw's picture
Training in progress, step 100
0560a09 verified
{
"_name_or_path": "facebook/dinov2-base",
"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": 768,
"id2label": {
"0": "Alternative",
"1": "Animecore",
"10": "Dark_Fantasy",
"11": "E-Girl",
"12": "Ethereal",
"13": "Fairycore",
"14": "Goblincore",
"15": "Goth",
"16": "Grunge",
"17": "Hypebeast",
"18": "Jersey_Shore",
"19": "Kawaii",
"2": "Art_Hoe",
"20": "Kidcore",
"21": "Lagenlook",
"22": "Lolita",
"23": "Mob_Wife",
"24": "Normcore",
"25": "Old_Money",
"26": "Pastel_Goth",
"27": "Preppy",
"28": "Punk",
"29": "Soft_Grunge",
"3": "Avant-garde",
"30": "Steampunk",
"31": "Tomato_Girl_Summer",
"32": "Twee",
"33": "VSCO",
"34": "Vaporwave",
"35": "Y2K",
"4": "Baddie",
"5": "Bohemian",
"6": "Cottagecore",
"7": "Country",
"8": "Cyberpunk",
"9": "Dark_Academia"
},
"image_size": 518,
"initializer_range": 0.02,
"label2id": {
"Alternative": "0",
"Animecore": "1",
"Art_Hoe": "2",
"Avant-garde": "3",
"Baddie": "4",
"Bohemian": "5",
"Cottagecore": "6",
"Country": "7",
"Cyberpunk": "8",
"Dark_Academia": "9",
"Dark_Fantasy": "10",
"E-Girl": "11",
"Ethereal": "12",
"Fairycore": "13",
"Goblincore": "14",
"Goth": "15",
"Grunge": "16",
"Hypebeast": "17",
"Jersey_Shore": "18",
"Kawaii": "19",
"Kidcore": "20",
"Lagenlook": "21",
"Lolita": "22",
"Mob_Wife": "23",
"Normcore": "24",
"Old_Money": "25",
"Pastel_Goth": "26",
"Preppy": "27",
"Punk": "28",
"Soft_Grunge": "29",
"Steampunk": "30",
"Tomato_Girl_Summer": "31",
"Twee": "32",
"VSCO": "33",
"Vaporwave": "34",
"Y2K": "35"
},
"layer_norm_eps": 1e-06,
"layerscale_value": 1.0,
"mlp_ratio": 4,
"model_type": "dinov2",
"num_attention_heads": 12,
"num_channels": 3,
"num_hidden_layers": 12,
"out_features": [
"stage12"
],
"out_indices": [
12
],
"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"
],
"torch_dtype": "float32",
"transformers_version": "4.41.2",
"use_swiglu_ffn": false
}