FruitsClassifier with Pytorch Lightning 🍉

On this mission, I convert my FruitsClassifier from Pytorch to Pytorch Lightning. I make use of a callback module, a checkpoint mode to retailer and save the mannequin state_dict. I conduct resume coaching, plot loss and accuracy from earlier trains. I test the accuracy for every class and make predictions on pictures from exterior the […]