How to use roylvzn/vit-cifar10 with timm:
import timm model = timm.create_model("hf_hub:roylvzn/vit-cifar10", pretrained=True)
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