Update README.md
Browse files
README.md
CHANGED
|
@@ -9,7 +9,7 @@ datasets:
|
|
| 9 |
- imagenet-1k
|
| 10 |
library_name: mlx-image
|
| 11 |
---
|
| 12 |
-
# vit_huge_patch14_518
|
| 13 |
|
| 14 |
A [Vision Transformer](https://arxiv.org/abs/2010.11929v2) image classification model. Weights are learned with [SWAG](https://arxiv.org/abs/2201.08371) on ImageNet-1k data.
|
| 15 |
|
|
|
|
| 9 |
- imagenet-1k
|
| 10 |
library_name: mlx-image
|
| 11 |
---
|
| 12 |
+
# vit_huge_patch14_518.swag_e2e
|
| 13 |
|
| 14 |
A [Vision Transformer](https://arxiv.org/abs/2010.11929v2) image classification model. Weights are learned with [SWAG](https://arxiv.org/abs/2201.08371) on ImageNet-1k data.
|
| 15 |
|