outputs / config.json
HimanshuWiai's picture
Training in progress, epoch 0
96c58e4 verified
{
"_name_or_path": "google/vit-base-patch16-224",
"architectures": [
"ViTForImageClassification"
],
"attention_probs_dropout_prob": 0.0,
"encoder_stride": 16,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 768,
"id2label": {
"0": "bacterial_blight",
"1": "bacterial_leaf_blight",
"2": "bacterial_leaf_streak",
"3": "bacterial_panicle_blight",
"4": "black_stem_borer",
"5": "blast",
"6": "brown_plant_hopper",
"7": "brown_spot",
"8": "downy_mildew",
"9": "false_smut",
"10": "gundhi_bug",
"11": "healthy",
"12": "hispa",
"13": "ladybird_beetle",
"14": "leaf_folder",
"15": "leaf_roller",
"16": "leaf_smut",
"17": "others",
"18": "tungro",
"19": "white_stem_borer",
"20": "yellow_stem_borer"
},
"image_size": 224,
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"bacterial_blight": 0,
"bacterial_leaf_blight": 1,
"bacterial_leaf_streak": 2,
"bacterial_panicle_blight": 3,
"black_stem_borer": 4,
"blast": 5,
"brown_plant_hopper": 6,
"brown_spot": 7,
"downy_mildew": 8,
"false_smut": 9,
"gundhi_bug": 10,
"healthy": 11,
"hispa": 12,
"ladybird_beetle": 13,
"leaf_folder": 14,
"leaf_roller": 15,
"leaf_smut": 16,
"others": 17,
"tungro": 18,
"white_stem_borer": 19,
"yellow_stem_borer": 20
},
"layer_norm_eps": 1e-12,
"model_type": "vit",
"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.46.0.dev0"
}