Classifying galaxy images with neural networks in PyTorch