all AI news
FruitsClassifier with Pytorch Lightning 🍉
June 27, 2022, 5:19 p.m. | Peter Bao
DEV Community dev.to
On this project, I convert my FruitsClassifier from Pytorch to Pytorch Lightning. I employ a callback module, a checkpoint mode to store and save the model state_dict. I conduct resume training, plot loss and accuracy from previous trains. I check the accuracy for each class and make predictions on images from outside the dataset.
✍️ Documentation:
- Convert my Pytorch model to Pytorch Lightning
- FruitsClassifier on Fruits 360
- Full Training on Kaggle | GoogleNet | 94% Acc
- …
deeplearning machinelearning python pytorch pytorch lightning
More from dev.to / DEV Community
Jobs in AI, ML, Big Data
AI Research Scientist
@ Vara | Berlin, Germany and Remote
Data Architect
@ University of Texas at Austin | Austin, TX
Data ETL Engineer
@ University of Texas at Austin | Austin, TX
Lead GNSS Data Scientist
@ Lurra Systems | Melbourne
Senior Data Engineer (m/f/d)
@ Project A Ventures | Berlin, Germany
Principle Research Scientist
@ Analog Devices | US, MA, Boston