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).
state_dict
pytorch_model.bin
vit_model.py
timm