Update README.md
Browse files
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 |
|