johnnv commited on
Commit
9c74948
1 Parent(s): 60ab71a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -3,7 +3,7 @@ license: apache-2.0
3
  pipeline_tag: image-classification
4
  ---
5
 
6
- Pytorch weights for Kornia ViT.
7
 
8
  Original weights from https://github.com/google-research/vision_transformer: This weight is based on the [Original ViT_L/16 pretrained on imagenet21k](https://storage.googleapis.com/vit_models/augreg/L_16-i21k-300ep-lr_0.001-aug_strong1-wd_0.1-do_0.0-sd_0.0.npz)
9
 
 
3
  pipeline_tag: image-classification
4
  ---
5
 
6
+ Pytorch weights for Kornia ViT converted from the original google JAX vision-transformer repo.
7
 
8
  Original weights from https://github.com/google-research/vision_transformer: This weight is based on the [Original ViT_L/16 pretrained on imagenet21k](https://storage.googleapis.com/vit_models/augreg/L_16-i21k-300ep-lr_0.001-aug_strong1-wd_0.1-do_0.0-sd_0.0.npz)
9