This repo contains some demo code when I'm learning pytorch from http://www.learnpytorch.io
Thanks to Daniel Bourke, really a great cource.
Git repo of the cource: https://github.com/mrdbourke/pytorch-deep-learning/tree/main
- Going Modular
- Vision Transformer
- Model Deployment
I ignored some image files and model files(.pth), to decrease the size of the repo, but run the code in this repo can reload/create them.