1aurent commited on
Commit
a68a75d
1 Parent(s): 630eca2

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "architecture": "resnet18",
3
  "num_classes": 2,
4
- "num_features": 2048,
5
  "pretrained_cfg": {
6
  "tag": "tiatoolbox-pcam",
7
  "custom_load": false,
 
1
  {
2
  "architecture": "resnet18",
3
  "num_classes": 2,
4
+ "num_features": 512,
5
  "pretrained_cfg": {
6
  "tag": "tiatoolbox-pcam",
7
  "custom_load": false,