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
@@ -13,7 +13,7 @@ A timm specific CoAtNet image classification model. Trained in `timm` on ImageNe
 
 ImageNet-1k training done on TPUs thanks to support of the [TRC](https://sites.research.google/trc/about/) program.
 
-### 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.
@@ -43,8 +43,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('coatnet_nano_rw_224.sw_in1k', pretrained=True)
 model = model.eval()
@@ -64,8 +65,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(
     'coatnet_nano_rw_224.sw_in1k',
@@ -82,12 +84,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, 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])
+
     print(o.shape)
 ```
 
@@ -97,8 +100,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(
     'coatnet_nano_rw_224.sw_in1k',
@@ -116,10 +120,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, 512, 7, 7) 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
@@ -227,7 +231,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:555f9ea6286c8c2aed425a96fd6f2b00308e3974e318b6daf3918e81c1838a6e
+size 60641304