Update model config and README

README.md

@@ -1,7 +1,7 @@
 ---
 tags:
-- image-classification
 - timm
+- image-classification
 library_name: timm
 license: apache-2.0
 datasets:
@@ -9,14 +9,14 @@ datasets:
 ---
 # Model card for vit_base_r50_s16_224.orig_in21k
 
-A ResNet - Vision Transformer (ViT) hybrid image classification model. Trained on ImageNet-21k in JAX by paper authors, ported to PyTorch by Ross Wightman.
+A ResNet - Vision Transformer (ViT) hybrid image classification model. Pretrained on ImageNet-21k in JAX by paper authors, ported to PyTorch by Ross Wightman. This model does not have a classification head, useful for features and fine-tune only.
 
 
 ## Model Details
 - **Model Type:** Image classification / feature backbone
 - **Model Stats:**
-  - Params (M): 114.7
-  - GMACs: 21.0
+  - Params (M): 97.9
+  - GMACs: 20.9
   - Activations (M): 27.9
   - Image size: 224 x 224
 - **Papers:**