1aurent
/

vit_base_patch16_224.owkin_pancancer_ft_lc25000_lung

Image Classification

Model card Files Files and versions Community

1aurent commited on Oct 23, 2023

Commit

72fe16a

•

1 Parent(s): ecd93b8

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -1,10 +1,11 @@
 {
   "architecture": "vit_base_patch16_224",
   "num_classes": 3,
   "num_features": 768,
   "global_pool": "token",
   "pretrained_cfg": {
-    "tag": "owkin_pancancer",
     "custom_load": false,
     "input_size": [
       3,
@@ -25,7 +26,6 @@
       0.224,
       0.225
     ],
-    "num_classes": 0,
     "pool_size": null,
     "first_conv": "patch_embed.proj",
     "classifier": "head"

 {
   "architecture": "vit_base_patch16_224",
   "num_classes": 3,
+  "label_names": ["bening", "adenocarcinomas", "squamous carcinomas"],
   "num_features": 768,
   "global_pool": "token",
   "pretrained_cfg": {
+    "tag": "owkin_pancancer_ft_lc25000_lung",
     "custom_load": false,
     "input_size": [
       3,
       0.224,
       0.225
     ],
     "pool_size": null,
     "first_conv": "patch_embed.proj",
     "classifier": "head"