Create a virtual environment: python3 -m venv .sparsity_env

Load the virtual environment: source .sparsity_env/bin/activate

Install dependencies: pip install -r requirements.txt

prune.py can run any pruning method & model. To save time, run sh ./prune.sh and keep track of any arguments in there.