1aurent commited on
Commit
5cfbdba
1 Parent(s): b134c8b

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +2 -2
config.json CHANGED
@@ -1,10 +1,11 @@
1
  {
2
  "architecture": "vit_base_patch16_224",
3
  "num_classes": 2,
 
4
  "num_features": 768,
5
  "global_pool": "token",
6
  "pretrained_cfg": {
7
- "tag": "owkin_pancancer",
8
  "custom_load": false,
9
  "input_size": [
10
  3,
@@ -25,7 +26,6 @@
25
  0.224,
26
  0.225
27
  ],
28
- "num_classes": 0,
29
  "pool_size": null,
30
  "first_conv": "patch_embed.proj",
31
  "classifier": "head"
 
1
  {
2
  "architecture": "vit_base_patch16_224",
3
  "num_classes": 2,
4
+ "label_names": ["bening", "adenocarcinomas"],
5
  "num_features": 768,
6
  "global_pool": "token",
7
  "pretrained_cfg": {
8
+ "tag": "owkin_pancancer_ft_lc25000_colon",
9
  "custom_load": false,
10
  "input_size": [
11
  3,
 
26
  0.224,
27
  0.225
28
  ],
 
29
  "pool_size": null,
30
  "first_conv": "patch_embed.proj",
31
  "classifier": "head"