Classification with PyTorch. - View it on GitHub