Dog Cat Classifier in PyTorch