Any best practices example on integrating a pretrained HuggingFace ViT into a pytorch lightning module?

Models
2

Hi,

I do have a notebook on fine-tuning ViT with PyTorch Lightning here: Transformers-Tutorials/Fine_tuning_the_Vision_Transformer_on_CIFAR_10_with_PyTorch_Lightning.ipynb at master · NielsRogge/Transformers-Tutorials · GitHub

1 Like