pytorch

Image classification on self generated dataset using PyTorch