How to train？

[LiJichen0114]

Hello, thank you very much for your work. I am eager to know how to train a ViT model from scratch on the ImageNet dataset. I haven’t seen the Python file for training. Thank you!

[GeneralSubhra]

you can check my repo,I have build that from scratch

[mahendran-narayanan]

Check this file for training the model in pytorch.
Use the model from this repository and follow the code in the link

PyTorch training