|
--- |
|
license: mit |
|
tags: |
|
- 'vit ' |
|
- image classification |
|
- ggml |
|
--- |
|
|
|
# Vision Transformer (ViT) models for image classification converted to ggml format |
|
|
|
[Available models](https://github.com/staghado/vit.cpp) |
|
|
|
| Model | Disk | Mem | SHA | |
|
| --- | --- | --- | --- | |
|
| tiny | 12 MB | ~20 MB | `25ce65ff60e08a1a5b486685b533d79718e74c0f` | |
|
| small | 45 MB | ~52 MB | `7a9f85340bd1a3dcd4275f46d5ee1db66649700e` | |
|
| base | 174 MB | ~179 MB | `a10d29628977fe27691edf55b7238f899b8c02eb` | |
|
| large | 610 MB | ~597 MB | `5f27087930f21987050188f9dc9eea75ac607214` | |
|
|
|
The models are pre-trained on ImageNet21k then finetuned on ImageNet1k |
|
with a patch size of 16 and an image size of 224. |
|
|
|
For more information, visit: |
|
|
|
https://github.com/staghado/vit.cpp |
|
|