Imported from https://github.com/zhangtj1996/lottery-ticket-hypothesis-Mxnet