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 @@ An official MaxViT image classification model. Pretrained in tensorflow on Image
 
 Ported from official Tensorflow implementation (https://github.com/google-research/maxvit) to PyTorch by Ross Wightman.
 
-### 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.
@@ -45,8 +45,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_base_tf_384.in21k_ft_in1k', pretrained=True)
 model = model.eval()
@@ -66,8 +67,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_base_tf_384.in21k_ft_in1k',
@@ -84,12 +86,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)
 ```
 
@@ -99,8 +102,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_base_tf_384.in21k_ft_in1k',
@@ -118,10 +122,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
@@ -229,7 +233,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:81cdb9d0c4697a0cee85e768bc45338ff0bdfd4c5e35058fb5d4cca75f58e100
+size 479337070