https://machinelearningmastery.com/building-a-convolutional-neural-network-in-pytorch/