This is an official source code for "Towards Data-Agnostic Pruning at Initialization: What Makes a Good Sparse Mask?", NeurIPS 2023.
Python >= 3.6, Pytorch >= 1.8, Cvxpy
sh scripts/run.sh
This source code is adopted from "Pruning neural networks without any data by iteratively conserving synaptic flow".