Train a neural network with a Gaussian Distributed noise injection regularization. Quantization-aware training has been adopted from "Aggressive Quantization: How to run MNIST on a 4 bit Neural Net using Pytorch" by Karanbir Chahal: https://medium.com/@karanbirchahal/aggressive-quantization-how-to-run-mnist-on-a-4-bit-neural-net-using-pytorch-5703f3faa599