Create a virtual environment conda env create -f environment.yml source activate pruning Run python download_model.py python pruning.py