Skip to content

Yaminik1996/CNNPruning

Repository files navigation

Team members:

  1. Nidhi Chandra
  2. Yamini Kashyap
  3. Indrajeet Nandy

Steps to run the project:

  1. Create a virtual environment where you can install the required libraries and run the pruned models:

     virtualenv -p python3 pruning
     source pruning/bin/activate
    
  2. 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.

About

Pruning of pretrained Resnet18 model using PyTorch pruning library.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors