Imported from https://github.com/vibhu444/alexnet-squeeze-mnist