my own datasets? train.py pre.py?

lucidrains / vit-pytorch

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

MIT License

20.27k stars 3.02k forks source link

Open henbucuoshanghai opened 2 years ago

larsonwu0220 commented 2 years ago

You can look examples/cats_and_dogs.ipynb Good luck!

beizhengren commented 1 year ago

Hi, after training, how to save weights?