kornia
/

vit_s16_augreg_i21k_r224

Image Classification

Model card Files Files and versions Community

johnnv commited on Feb 9

Commit

26743c9

•

1 Parent(s): f120726

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -5,7 +5,8 @@ pipeline_tag: image-classification
 Pytorch weights for Kornia ViT converted from the original google JAX vision-transformer repo.
-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/S_16-i21k-300ep-lr_0.001-aug_light1-wd_0.03-do_0.0-sd_0.0.npz)
 Weights converted to PyTorch for Kornia ViT implementation (by [@gau-nernst](https://github.com/gau-nernst) in [kornia/kornia#2786](https://github.com/kornia/kornia/pull/2786#discussion_r1482339811))
 <details>

 Pytorch weights for Kornia ViT converted from the original google JAX vision-transformer repo.
+Original weights from https://github.com/google-research/vision_transformer: This weight is based on the
+[Original ViT_S/16 pretrained on imagenet21k](https://storage.googleapis.com/vit_models/augreg/S_16-i21k-300ep-lr_0.001-aug_light1-wd_0.03-do_0.0-sd_0.0.npz)
 Weights converted to PyTorch for Kornia ViT implementation (by [@gau-nernst](https://github.com/gau-nernst) in [kornia/kornia#2786](https://github.com/kornia/kornia/pull/2786#discussion_r1482339811))
 <details>