There's a second part to my Image Classification task tutorial using Pytorch. Please check it out for a complete review!
https://dev.to/qbaocaca/full-pytorch-training-pipeline-on-image-classification-task-part2--185o