hoanbklucky's picture
Training in progress, epoch 1
96a674a verified
{
"_name_or_path": "facebook/dinov2-small-imagenet1k-1-layer",
"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": 384,
"id2label": {
"0": "abyssinian",
"1": "american_bulldog",
"2": "american_pit_bull_terrier",
"3": "basset_hound",
"4": "beagle",
"5": "bengal",
"6": "birman",
"7": "bombay",
"8": "boxer",
"9": "british_shorthair",
"10": "chihuahua",
"11": "egyptian_mau",
"12": "english_cocker_spaniel",
"13": "english_setter",
"14": "german_shorthaired",
"15": "great_pyrenees",
"16": "havanese",
"17": "japanese_chin",
"18": "keeshond",
"19": "leonberger",
"20": "maine_coon",
"21": "miniature_pinscher",
"22": "newfoundland",
"23": "persian",
"24": "pomeranian",
"25": "pug",
"26": "ragdoll",
"27": "russian_blue",
"28": "saint_bernard",
"29": "samoyed",
"30": "scottish_terrier",
"31": "shiba_inu",
"32": "siamese",
"33": "sphynx",
"34": "staffordshire_bull_terrier",
"35": "wheaten_terrier",
"36": "yorkshire_terrier"
},
"image_size": 518,
"initializer_range": 0.02,
"label2id": {
"abyssinian": 0,
"american_bulldog": 1,
"american_pit_bull_terrier": 2,
"basset_hound": 3,
"beagle": 4,
"bengal": 5,
"birman": 6,
"bombay": 7,
"boxer": 8,
"british_shorthair": 9,
"chihuahua": 10,
"egyptian_mau": 11,
"english_cocker_spaniel": 12,
"english_setter": 13,
"german_shorthaired": 14,
"great_pyrenees": 15,
"havanese": 16,
"japanese_chin": 17,
"keeshond": 18,
"leonberger": 19,
"maine_coon": 20,
"miniature_pinscher": 21,
"newfoundland": 22,
"persian": 23,
"pomeranian": 24,
"pug": 25,
"ragdoll": 26,
"russian_blue": 27,
"saint_bernard": 28,
"samoyed": 29,
"scottish_terrier": 30,
"shiba_inu": 31,
"siamese": 32,
"sphynx": 33,
"staffordshire_bull_terrier": 34,
"wheaten_terrier": 35,
"yorkshire_terrier": 36
},
"layer_norm_eps": 1e-06,
"layerscale_value": 1.0,
"mlp_ratio": 4,
"model_type": "dinov2",
"num_attention_heads": 6,
"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.46.3",
"use_swiglu_ffn": false
}