Update config.json

config.json

@@ -3,35 +3,41 @@
   "num_classes": 2,
   "num_features": 2048,
   "pretrained_cfg": {
-    "tag": "tv_in1k",
+    "tag": "tiatoolbox-pcam",
     "custom_load": false,
     "input_size": [
       3,
-      224,
-      224
+      96,
+      96
+    ],
+    "test_input_size": [
+      3,
+      96,
+      96
     ],
     "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": 2,
     "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"
   }
 }