deit-ena24 / config.json
mbiarreta's picture
Training in progress, step 100
69e2d95 verified
{
"architectures": [
"DeiTForImageClassification"
],
"attention_probs_dropout_prob": 0.0,
"encoder_stride": 16,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 768,
"id2label": {
"0": "American Black Bear",
"1": "American Crow",
"10": "Eastern Fox Squirrel",
"11": "Eastern Gray Squirrel",
"12": "Grey Fox",
"13": "Horse",
"14": "Northern Raccoon",
"15": "Red Fox",
"16": "Striped Skunk",
"17": "Vehicle",
"18": "Virginia Opossum",
"19": "White_Tailed_Deer",
"2": "Bird",
"20": "Wild Turkey",
"21": "Woodchuck",
"3": "Bobcat",
"4": "Chicken",
"5": "Coyote",
"6": "Dog",
"7": "Domestic Cat",
"8": "Eastern Chipmunk",
"9": "Eastern Cottontail"
},
"image_size": 224,
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"American Black Bear": "0",
"American Crow": "1",
"Bird": "2",
"Bobcat": "3",
"Chicken": "4",
"Coyote": "5",
"Dog": "6",
"Domestic Cat": "7",
"Eastern Chipmunk": "8",
"Eastern Cottontail": "9",
"Eastern Fox Squirrel": "10",
"Eastern Gray Squirrel": "11",
"Grey Fox": "12",
"Horse": "13",
"Northern Raccoon": "14",
"Red Fox": "15",
"Striped Skunk": "16",
"Vehicle": "17",
"Virginia Opossum": "18",
"White_Tailed_Deer": "19",
"Wild Turkey": "20",
"Woodchuck": "21"
},
"layer_norm_eps": 1e-12,
"model_type": "deit",
"num_attention_heads": 12,
"num_channels": 3,
"num_hidden_layers": 12,
"patch_size": 16,
"pooler_act": "tanh",
"pooler_output_size": 768,
"problem_type": "single_label_classification",
"qkv_bias": true,
"torch_dtype": "float32",
"transformers_version": "4.52.3"
}