Update model config and README

README.md

@@ -2,7 +2,7 @@
 tags:
 - image-classification
 - timm
-library_tag: timm
+library_name: timm
 license: apache-2.0
 datasets:
 - imagenet-1k
@@ -14,7 +14,7 @@ A timm specific MaxViT (w/ a MLP Log-CPB (continuous log-coordinate relative pos
 
 ImageNet-12k pretraining and ImageNet-1k fine-tuning performed on 8x GPU [Lambda Labs](https://lambdalabs.com/) cloud instances..
 
-### Model Variants in [maxxvit.py](https://github.com/rwightman/pytorch-image-models/blob/main/timm/models/maxxvit.py)
+### Model Variants in [maxxvit.py](https://github.com/huggingface/pytorch-image-models/blob/main/timm/models/maxxvit.py)
 
 MaxxViT covers a number of related model architectures that share a common structure including:
 - CoAtNet - Combining MBConv (depthwise-separable) convolutional blocks in early stages with self-attention transformer blocks in later stages.
@@ -46,8 +46,9 @@ from urllib.request import urlopen
 from PIL import Image
 import timm
 
-img = Image.open(
-    urlopen('https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/beignets-task-guide.png'))
+img = Image.open(urlopen(
+    'https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/beignets-task-guide.png'
+))
 
 model = timm.create_model('maxvit_rmlp_base_rw_384.sw_in12k_ft_in1k', pretrained=True)
 model = model.eval()
@@ -67,8 +68,9 @@ from urllib.request import urlopen
 from PIL import Image
 import timm
 
-img = Image.open(
-    urlopen('https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/beignets-task-guide.png'))
+img = Image.open(urlopen(
+    'https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/beignets-task-guide.png'
+))
 
 model = timm.create_model(
     'maxvit_rmlp_base_rw_384.sw_in12k_ft_in1k',
@@ -85,12 +87,13 @@ 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, 128, 192, 192])
-    #  torch.Size([1, 128, 96, 96])
-    #  torch.Size([1, 256, 48, 48])
-    #  torch.Size([1, 512, 24, 24])
-    #  torch.Size([1, 1024, 12, 12])
+    # e.g.:
+    #  torch.Size([1, 64, 192, 192])
+    #  torch.Size([1, 96, 96, 96])
+    #  torch.Size([1, 192, 48, 48])
+    #  torch.Size([1, 384, 24, 24])
+    #  torch.Size([1, 768, 12, 12])
+
     print(o.shape)
 ```
 
@@ -100,8 +103,9 @@ from urllib.request import urlopen
 from PIL import Image
 import timm
 
-img = Image.open(
-    urlopen('https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/beignets-task-guide.png'))
+img = Image.open(urlopen(
+    'https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/beignets-task-guide.png'
+))
 
 model = timm.create_model(
     'maxvit_rmlp_base_rw_384.sw_in12k_ft_in1k',
@@ -119,10 +123,10 @@ 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 (ie.e a (batch_size, num_features, H, W) tensor
+# output is unpooled, a (1, 768, 12, 12) shaped tensor
 
 output = model.forward_head(output, pre_logits=True)
-# output is (batch_size, num_features) tensor
+# output is a (1, num_features) shaped tensor
 ```
 
 ## Model Comparison
@@ -230,7 +234,7 @@ output = model.forward_head(output, pre_logits=True)
   publisher = {GitHub},
   journal = {GitHub repository},
   doi = {10.5281/zenodo.4414861},
-  howpublished = {\url{https://github.com/rwightman/pytorch-image-models}}
+  howpublished = {\url{https://github.com/huggingface/pytorch-image-models}}
 }
 ```
 ```bibtex

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:36937552a45c431fbd34e77456c0c2f44c0874632bed47f28c650c1a0fbf8821
+size 465234858