UCSCPhytoViT83 / config.json
Patrick Daniel
Changed model weight
986b17c
{
"architectures": [
"ViTForImageClassification"
],
"attention_probs_dropout_prob": 0.0,
"encoder_stride": 16,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 768,
"id2label": {
"0": "Akashiwo",
"1": "Alexandrium",
"2": "Amylax_Gonyaulax_Protoceratium",
"3": "Asterionellopsis",
"4": "Asteromphalus",
"5": "Bad_setae",
"6": "Centric",
"7": "Ceratium_divaricatum",
"8": "Ceratium_furca",
"9": "Ceratium_lineatum",
"10": "Chaetoceros",
"11": "Ciliate_large",
"12": "Ciliate_large_2",
"13": "Ciliate_other_morpho_1",
"14": "Clusterflagellate_morpho_1",
"15": "Clusterflagellate_morpho_2",
"16": "Corethron",
"17": "Cryptophyte",
"18": "Cylindrotheca",
"19": "Detonula_Cerataulina_Lauderia",
"20": "Detritus",
"21": "Detritus_infection",
"22": "Dictyocha",
"23": "Dinoflagellate_cyst",
"24": "Dinoflagellate_morpho_1",
"25": "Dinoflagellate_morpho_2",
"26": "Dinophysis",
"27": "Ditylum",
"28": "Entomoneis",
"29": "Eucampia",
"30": "Euglenoid",
"31": "Flagellate_morpho_1",
"32": "Flagellate_morpho_2",
"33": "Flagellate_morpho_3",
"34": "Flagellate_nano_1",
"35": "Flagellate_nano_2",
"36": "Fragilariopsis",
"37": "Guinardia_Dactyliosolen",
"38": "Gymnodinium",
"39": "Gyrodinium",
"40": "Gyrosigma",
"41": "Haptophyte_prymnesium",
"42": "Hemiaulus",
"43": "Hemiselmis",
"44": "Heterocapsa_long",
"45": "Heterocapsa_rotundata",
"46": "Heterocapsa_triquetra",
"47": "Heterosigma_akashiwo",
"48": "Laboea",
"49": "Leptocylindrus",
"50": "Margalefidinium",
"51": "Mesodinium",
"52": "Nano_cluster",
"53": "Nano_p_white",
"54": "Noctiluca",
"55": "Odontella",
"56": "Pennate",
"57": "Pennate_Tropidoneis",
"58": "Pennate_Unknown",
"59": "Pennate_small",
"60": "Peridinium",
"61": "Phaeocystis",
"62": "Pleurosigma",
"63": "Polykrikos",
"64": "Proboscia",
"65": "Prorocentrum_narrow",
"66": "Prorocentrum_wide",
"67": "Pseudo-nitzschia",
"68": "Pyramimonas",
"69": "Rhizosolenia",
"70": "Scrippsiella",
"71": "Skeletonema",
"72": "Spiky_pacman",
"73": "Stombidinium_morpho_1",
"74": "Strombidinum_morpho_2",
"75": "Thalassionema",
"76": "Thalassiosira",
"77": "Tiarina",
"78": "Tontonia",
"79": "Torodinium",
"80": "Tropidoneis",
"81": "Vicicitus",
"82": "haptophyte_ucynA_host"
},
"image_size": 224,
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"Akashiwo": 0,
"Alexandrium": 1,
"Amylax_Gonyaulax_Protoceratium": 2,
"Asterionellopsis": 3,
"Asteromphalus": 4,
"Bad_setae": 5,
"Centric": 6,
"Ceratium_divaricatum": 7,
"Ceratium_furca": 8,
"Ceratium_lineatum": 9,
"Chaetoceros": 10,
"Ciliate_large": 11,
"Ciliate_large_2": 12,
"Ciliate_other_morpho_1": 13,
"Clusterflagellate_morpho_1": 14,
"Clusterflagellate_morpho_2": 15,
"Corethron": 16,
"Cryptophyte": 17,
"Cylindrotheca": 18,
"Detonula_Cerataulina_Lauderia": 19,
"Detritus": 20,
"Detritus_infection": 21,
"Dictyocha": 22,
"Dinoflagellate_cyst": 23,
"Dinoflagellate_morpho_1": 24,
"Dinoflagellate_morpho_2": 25,
"Dinophysis": 26,
"Ditylum": 27,
"Entomoneis": 28,
"Eucampia": 29,
"Euglenoid": 30,
"Flagellate_morpho_1": 31,
"Flagellate_morpho_2": 32,
"Flagellate_morpho_3": 33,
"Flagellate_nano_1": 34,
"Flagellate_nano_2": 35,
"Fragilariopsis": 36,
"Guinardia_Dactyliosolen": 37,
"Gymnodinium": 38,
"Gyrodinium": 39,
"Gyrosigma": 40,
"Haptophyte_prymnesium": 41,
"Hemiaulus": 42,
"Hemiselmis": 43,
"Heterocapsa_long": 44,
"Heterocapsa_rotundata": 45,
"Heterocapsa_triquetra": 46,
"Heterosigma_akashiwo": 47,
"Laboea": 48,
"Leptocylindrus": 49,
"Margalefidinium": 50,
"Mesodinium": 51,
"Nano_cluster": 52,
"Nano_p_white": 53,
"Noctiluca": 54,
"Odontella": 55,
"Pennate": 56,
"Pennate_Tropidoneis": 57,
"Pennate_Unknown": 58,
"Pennate_small": 59,
"Peridinium": 60,
"Phaeocystis": 61,
"Pleurosigma": 62,
"Polykrikos": 63,
"Proboscia": 64,
"Prorocentrum_narrow": 65,
"Prorocentrum_wide": 66,
"Pseudo-nitzschia": 67,
"Pyramimonas": 68,
"Rhizosolenia": 69,
"Scrippsiella": 70,
"Skeletonema": 71,
"Spiky_pacman": 72,
"Stombidinium_morpho_1": 73,
"Strombidinum_morpho_2": 74,
"Thalassionema": 75,
"Thalassiosira": 76,
"Tiarina": 77,
"Tontonia": 78,
"Torodinium": 79,
"Tropidoneis": 80,
"Vicicitus": 81,
"haptophyte_ucynA_host": 82
},
"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.54.0"
}