Data This implementation supports cifar10/cifar100/imagenet Dependencies tensorflow version 1.4.1 Training Train imagenet model using gpu: python main.py --model alexnet_QNN --save alexnet_QNN --dataset imagenet --gpu True