vit-cifar10 / README.md
roylvzn's picture
update README.md
5916238 verified
metadata
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

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