Imported from https://github.com/guoxiaolu/model_compression