Update config.json

config.json

@@ -1,37 +1,44 @@
 {
   "architecture": "wide_resnet101_2",
   "num_classes": 9,
+  "label_names": ["BACK", "NORM", "DEB", "TUM", "ADI", "MUC", "MUS", "STR", "LYM"],
   "num_features": 2048,
   "pretrained_cfg": {
-    "tag": "tv_in1k",
+    "tag": "tiatoolbox-kather100k",
     "custom_load": false,
     "input_size": [
       3,
       224,
       224
     ],
+    "test_input_size": [
+      3,
+      224,
+      224
+    ],
     "fixed_input_size": false,
-    "interpolation": "bilinear",
-    "crop_pct": 0.875,
+    "interpolation": "bicubic",
+    "crop_pct": 1.0,
+    "test_crop_pct": 1.0,
     "crop_mode": "center",
     "mean": [
-      0.485,
-      0.456,
-      0.406
+      0.0,
+      0.0,
+      0.0
     ],
     "std": [
-      0.229,
-      0.224,
-      0.225
+      1.0,
+      1.0,
+      1.0
     ],
-    "num_classes": 1000,
+    "num_classes": 9,
     "pool_size": [
       7,
       7
     ],
     "first_conv": "conv1",
     "classifier": "fc",
-    "license": "bsd-3-clause",
-    "origin_url": "https://github.com/pytorch/vision"
+    "origin_url": "https://github.com/huggingface/pytorch-image-models",
+    "paper_ids": "arXiv:2110.00476"
   }
 }