Imported from https://github.com/yeshaokai/Robustness-Aware-Pruning-ADMM