An Interest In:
Web News this Week
- April 20, 2024
- April 19, 2024
- April 18, 2024
- April 17, 2024
- April 16, 2024
- April 15, 2024
- April 14, 2024
June 27, 2022 05:19 pm GMT
Original Link: https://dev.to/qbaocaca/fruitsclassifier-with-pytorch-lightning-49gi
FruitsClassifier with Pytorch Lightning
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
- GitHub Page
Training Epochs
Visualize Loss and Accuracy
Visualize model performance on test dataset
Accuracy of each class
Image Predictions from outside the Dataset
Accuracy of each class: green(correct predictions), red(incorrect predictions)
Original Link: https://dev.to/qbaocaca/fruitsclassifier-with-pytorch-lightning-49gi
Share this article:
Tweet
View Full Article
Dev To
An online community for sharing and discovering great ideas, having debates, and making friendsMore About this Source Visit Dev To