Pytorch

Image classification on self generated dataset using PyTorch