Update model config and README

README.md

@@ -19,9 +19,9 @@ A collection of hparams (timm .yaml config files) for this training series can b
 - **Model Type:** Image classification / feature backbone
 - **Model Stats:**
   - Params (M): 21.8
-  - GMACs: 3.9
-  - Activations (M): 4.5
-  - Image size: train = 224 x 224, test = 448 x 448
+  - GMACs: 10.5
+  - Activations (M): 12.1
+  - Image size: train = 384 x 348, test = 448 x 448
 - **Dataset:** ImageNet-1k
 - **Papers:**
   - PyTorch Image Models: https://github.com/huggingface/pytorch-image-models
@@ -78,11 +78,11 @@ output = model(transforms(img).unsqueeze(0))  # unsqueeze single image into batc
 for o in output:
     # print shape of each feature map in output
     # e.g.:
-    #  torch.Size([1, 64, 112, 112])
-    #  torch.Size([1, 64, 56, 56])
-    #  torch.Size([1, 128, 28, 28])
-    #  torch.Size([1, 256, 14, 14])
-    #  torch.Size([1, 512, 7, 7])
+    #  torch.Size([1, 64, 192, 174])
+    #  torch.Size([1, 64, 96, 87])
+    #  torch.Size([1, 128, 48, 44])
+    #  torch.Size([1, 256, 24, 22])
+    #  torch.Size([1, 512, 12, 11])
 
     print(o.shape)
 ```
@@ -113,7 +113,7 @@ output = model(transforms(img).unsqueeze(0))  # output is (batch_size, num_featu
 # or equivalently (without needing to set num_classes=0)
 
 output = model.forward_features(transforms(img).unsqueeze(0))
-# output is unpooled, a (1, 512, 7, 7) shaped tensor
+# output is unpooled, a (1, 512, 12, 11) shaped tensor
 
 output = model.forward_head(output, pre_logits=True)
 # output is a (1, num_features) shaped tensor

config.json

@@ -7,8 +7,8 @@
     "custom_load": false,
     "input_size": [
       3,
-      224,
-      224
+      384,
+      348
     ],
     "test_input_size": [
       3,