Train a Vision Transformer (ViT) for custom image classification with data augmentation, transfer learning, and ONNX export.
Organize your images in class-named folders. Select a pretrained ViT base model matching your compute budget. Start with frozen backbone, then unfreeze for full fine-tuning if needed.
Initial release
Sign in and download this prompt to leave a review.