- Nidhi Chandra
- Yamini Kashyap
- Indrajeet Nandy
-
Create a virtual environment where you can install the required libraries and run the pruned models:
virtualenv -p python3 pruning source pruning/bin/activate -
Run the shell script run.sh, which downloads the model and also runs load_and_test.py
source run.sh
Find documentation for this code at docs/documentation.pdf
Credits to https://github.com/huyvnphan/PyTorch_CIFAR10 for setting up pretrained model on CIFAR10 data.