{ "_name_or_path": "ErnestBeckham/MulticancerViT", "architectures": ["ViTForImageClassification"], "id2label": { "0": "cervix_koc", "1": "cervix_dyk", "2": "cervix_pab", "3": "cervix_sfi", "4": "cervix_mep", "5": "colon_bnt", "6": "colon_aca", "7": "lung_aca", "8": "lung_bnt", "9": "lung_scc", "10": "oral_scc", "11": "oral_normal", "12":"kidney_tumor", "13":"kidney_normal", "14":"breast_benign", "15":"breast_malignant", "16":"lymph_fl", "17":"lymph_cll", "18":"lymph_mcl", "19":"brain_tumor", "20":"brain_glioma", "21":"brain_menin" }, "model_type": "vit", "lr":1e-4, "num_layers":12, "hidden_dim":512, "mlp_dim":12, "dropout":0.1 }