{ | |
"_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": "Aishwarya_Rai", | |
"1": "Akhilesh_Yadav", | |
"10": "MS_Dhoni", | |
"11": "Mahatma_Gandhi", | |
"12": "Mamata_Banerjee", | |
"13": "Mukesh_Ambani", | |
"14": "Narendra_Modi", | |
"15": "Rahul_Gandhi", | |
"16": "Ranveer_Singh", | |
"17": "Ratan_Tata", | |
"18": "Sachin_Tendulkar", | |
"19": "Shahrukh_Khan", | |
"2": "Akshay_Kumar", | |
"20": "Sonia_Gandhi", | |
"21": "Sundar_Pichai", | |
"22": "Virat_Kohli", | |
"23": "Yogi_Adityanath", | |
"3": "Amit_Saha", | |
"4": "Arijit_Singh", | |
"5": "Arvind_Kejriwal", | |
"6": "Baba_Ramdev", | |
"7": "Gautam_Adani", | |
"8": "Kishore_Kumar", | |
"9": "Lata_Mangeshkar" | |
}, | |
"image_size": 224, | |
"initializer_range": 0.02, | |
"intermediate_size": 3072, | |
"label2id": { | |
"Aishwarya_Rai": "0", | |
"Akhilesh_Yadav": "1", | |
"Akshay_Kumar": "2", | |
"Amit_Saha": "3", | |
"Arijit_Singh": "4", | |
"Arvind_Kejriwal": "5", | |
"Baba_Ramdev": "6", | |
"Gautam_Adani": "7", | |
"Kishore_Kumar": "8", | |
"Lata_Mangeshkar": "9", | |
"MS_Dhoni": "10", | |
"Mahatma_Gandhi": "11", | |
"Mamata_Banerjee": "12", | |
"Mukesh_Ambani": "13", | |
"Narendra_Modi": "14", | |
"Rahul_Gandhi": "15", | |
"Ranveer_Singh": "16", | |
"Ratan_Tata": "17", | |
"Sachin_Tendulkar": "18", | |
"Shahrukh_Khan": "19", | |
"Sonia_Gandhi": "20", | |
"Sundar_Pichai": "21", | |
"Virat_Kohli": "22", | |
"Yogi_Adityanath": "23" | |
}, | |
"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.24.0.dev0" | |
} | |