1aurent
/

vit_base_patch16_224.owkin_pancancer_ft_kather2016

Image Classification

Model card Files Files and versions Metrics Training metrics Community

1aurent commited on Oct 27, 2023

Commit

78af52b

•

1 Parent(s): 7277afc

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": 8,
   "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": 8,
+  "label_names": ["adipose", "complex", "debris", "empty", "lympho", "mucosa", "stroma", "tumor"],
   "num_features": 768,
   "global_pool": "token",
   "pretrained_cfg": {
+    "tag": "owkin_pancancer_ft_kather2016",
     "custom_load": false,
     "input_size": [
       3,
       0.224,
       0.225
     ],
     "pool_size": null,
     "first_conv": "patch_embed.proj",
     "classifier": "head"