M181_multiclass / config.json
Veit's picture
M181-trained dual-ViT (ViT_custom_size_sensitive_v5), test acc 96.17%
a995a57 verified
Raw History Blame Contribute Delete
1.33 kB
{
"architectures": [
"ViTForImageClassification"
],
"attention_probs_dropout_prob": 0.0,
"encoder_stride": 16,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 768,
"id2label": {
"0": "Asteroidea larvae",
"1": "Bacillariophyceae",
"10": "Thaliacea",
"11": "Cnidaria<Metazoa",
"12": "Eumalacostraca",
"2": "Chaetognatha",
"3": "Ctenophora_Metazoa",
"4": "Noctiluca sp.",
"5": "Pyrocystis",
"6": "Rhizaria",
"7": "Trichodesmium",
"8": "Appendicularia",
"9": "Copepoda"
},
"image_size": 224,
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"Appendicularia": "8",
"Asteroidea larvae": "0",
"Bacillariophyceae": "1",
"Chaetognatha": "2",
"Cnidaria<Metazoa": "11",
"Copepoda": "9",
"Ctenophora_Metazoa": "3",
"Eumalacostraca": "12",
"Noctiluca sp.": "4",
"Pyrocystis": "5",
"Rhizaria": "6",
"Thaliacea": "10",
"Trichodesmium": "7"
},
"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.52.4"
}