Update README.md

manogna-s · Nov 4, 2020 · 237d9e5 · 237d9e5
1 parent 95972e6
commit 237d9e5
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -12,9 +12,11 @@ Vision Transformer achieve State-of-the-Art in image recognition task with stand
 
 ## Usage
 ### 1. Download Pre-trained model
-* [Available models](https://console.cloud.google.com/storage/vit_models/): ViT-B_16(85.8M), ViT-B_32(87.5M), ViT-L_32(305.5M), ViT-H_14(630.8M)
-  * imagenet21k pre-train: ViT-B_16, ViT-B_32, ViT-L_32, ViT-H_14
-  * imagenet21k pre-train + imagenet2012 fine-tuning: ViT-B_16-224, ViT-B_16, ViT-B_32, ViT-L_16-224, ViT-L_16, ViT-L_32
+* [Available models](https://console.cloud.google.com/storage/vit_models/): ViT-B_16(**85.8M**), ViT-B_32(**87.5M**), ViT-L_32(**305.5M**), ViT-H_14(**630.8M**)
+  * imagenet21k pre-train models
+    * ViT-B_16, ViT-B_32, ViT-L_32, ViT-H_14
+  * imagenet21k pre-train + imagenet2012 fine-tuned models
+    * ViT-B_16-224, ViT-B_16, ViT-B_32, ViT-L_16-224, ViT-L_16, ViT-L_32
 ```
 # imagenet21k pre-train
 wget https://storage.googleapis.com/vit_models/imagenet21k/{MODEL_NAME}.npz