fish_classification / config.json
JJNeila's picture
Upload ViTForImageClassification
3bd9293 verified
{
"architectures": [
"ViTForImageClassification"
],
"attention_probs_dropout_prob": 0.0,
"encoder_stride": 16,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 768,
"id2label": {
"0": "Bangus",
"1": "Big Head Carp",
"10": "Gourami",
"11": "Grass Carp",
"12": "Green Spotted Puffer",
"13": "Indian Carp",
"14": "Indo-Pacific Tarpon",
"15": "Jaguar Gapote",
"16": "Janitor Fish",
"17": "Knifefish",
"18": "Long-Snouted Pipefish",
"19": "Mosquito Fish",
"2": "Black Spotted Barb",
"20": "Mudfish",
"21": "Mullet",
"22": "Pangasius",
"23": "Perch",
"24": "Scat Fish",
"25": "Silver Barb",
"26": "Silver Carp",
"27": "Silver Perch",
"28": "Snakehead",
"29": "Tenpounder",
"3": "Catfish",
"30": "Tilapia",
"4": "Climbing Perch",
"5": "Fourfinger Threadfin",
"6": "Freshwater Eel",
"7": "Glass Perchlet",
"8": "Goby",
"9": "Gold Fish"
},
"image_size": 224,
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"Bangus": "0",
"Big Head Carp": "1",
"Black Spotted Barb": "2",
"Catfish": "3",
"Climbing Perch": "4",
"Fourfinger Threadfin": "5",
"Freshwater Eel": "6",
"Glass Perchlet": "7",
"Goby": "8",
"Gold Fish": "9",
"Gourami": "10",
"Grass Carp": "11",
"Green Spotted Puffer": "12",
"Indian Carp": "13",
"Indo-Pacific Tarpon": "14",
"Jaguar Gapote": "15",
"Janitor Fish": "16",
"Knifefish": "17",
"Long-Snouted Pipefish": "18",
"Mosquito Fish": "19",
"Mudfish": "20",
"Mullet": "21",
"Pangasius": "22",
"Perch": "23",
"Scat Fish": "24",
"Silver Barb": "25",
"Silver Carp": "26",
"Silver Perch": "27",
"Snakehead": "28",
"Tenpounder": "29",
"Tilapia": "30"
},
"layer_norm_eps": 1e-12,
"model_type": "vit",
"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.53.0"
}