sruthis's picture
Training in progress, epoch 0
5346f15 verified
{
"_name_or_path": "facebook/deit-base-distilled-patch16-224",
"architectures": [
"DeiTForImageClassification"
],
"attention_probs_dropout_prob": 0.0,
"encoder_stride": 16,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 768,
"id2label": {
"0": "2-4D_Upper",
"1": "Bacterial_Spot_Upper",
"2": "Cold_Injury_Upper",
"3": "Early_Blight_Upper",
"4": "Healthy_Upperside",
"5": "Little_Leaf_Upperside",
"6": "Nutritional_Disorder_Upper",
"7": "Spider_Mite_Upper",
"8": "TSW_Upper",
"9": "TYLC_Upper"
},
"image_size": 224,
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"2-4D_Upper": "0",
"Bacterial_Spot_Upper": "1",
"Cold_Injury_Upper": "2",
"Early_Blight_Upper": "3",
"Healthy_Upperside": "4",
"Little_Leaf_Upperside": "5",
"Nutritional_Disorder_Upper": "6",
"Spider_Mite_Upper": "7",
"TSW_Upper": "8",
"TYLC_Upper": "9"
},
"layer_norm_eps": 1e-12,
"model_type": "deit",
"num_attention_heads": 12,
"num_channels": 3,
"num_hidden_layers": 12,
"patch_size": 16,
"problem_type": "single_label_classification",
"qkv_bias": true,
"torch_dtype": "float32",
"transformers_version": "4.35.2"
}