vit-cifar10 / README.md
roylvzn's picture
update README.md
5916238 verified
---
library_name: timm
license: apache-2.0
datasets:
- cifar10
tags:
- vision
- image-classification
- cifar10
- vit
model-index:
- name: vit-cifar10
results:
- task: {type: image-classification}
dataset: {name: CIFAR-10, type: cifar10}
metrics:
- type: accuracy
value: 0.95 # replace with your test accuracy
---
# ViT Base (patch16, 224) fine-tuned on CIFAR-10
Trained on CIFAR-10 (10 classes). Weights saved as a plain PyTorch `state_dict` (`pytorch_model.bin`).
Architecture is defined in `vit_model.py` (uses `timm`).