Ref: https://xmfbit.github.io/2017/03/04/pytorch-mnist-example/ Kaggle: Digit Recognizer Implemented with PyTorch 0.4.0