Image-Classification Cat - Dog Image Classifier I've limited resources so restricted epochs to 25 only...If you have a good GPU like Nvidia GTX1080, 100 epochs is a good number. Soon,this project will be available in Pytorch too!!!