File size: 747 Bytes
162458f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 | {
"model_type": "retvim",
"architectures": [
"RetViM"
],
"framework": "pytorch",
"format": "safetensors",
"backbone": "vit_base_patch16_224.augreg_in21k_ft_in1k",
"hybrid_blocks": 2,
"ssm_backend": "native-mamba-1-reference",
"feature_dim": 768,
"d_state": 16,
"d_conv": 4,
"expand": 2,
"classifier_dims": [
768,
512,
256,
4
],
"num_labels": 4,
"id2label": {
"0": "CNV",
"1": "DME",
"2": "DRUSEN",
"3": "NORMAL"
},
"label2id": {
"CNV": 0,
"DME": 1,
"DRUSEN": 2,
"NORMAL": 3
},
"parameter_count": 101223557,
"torch_dtype": "float32",
"note": "Custom PyTorch implementation from the linked GitHub repository; not a Transformers AutoModel checkpoint."
}
|