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`). | |