riccardomusmeci
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -11,7 +11,7 @@ library_name: mlx-image
|
|
11 |
---
|
12 |
# Model card for vit_base_patch16_224.swag_lin
|
13 |
|
14 |
-
A
|
15 |
|
16 |
Disclaimer: This is a porting of the torchvision model weights to Apple MLX Framework.
|
17 |
|
|
|
11 |
---
|
12 |
# Model card for vit_base_patch16_224.swag_lin
|
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 |
|
16 |
Disclaimer: This is a porting of the torchvision model weights to Apple MLX Framework.
|
17 |
|