/Pruning-PGP

Implementation of PGP Pruning for CNN

Primary LanguagePython

Description:

This is the implementation of the paper: An Improved Trade-off Between Accuracy and Complexitywith Progressive Gradient Pruning

Requirements

In order to use this repo, you need to install all the requirements in requirements.txt

How-to-use

You can run pgp_target_expo.py in order to use the PGP techiques.

You can run pgp_real_target_expo.py in order to use the RPGP techiques.

For changing hyper-parameters, you can directly change them in the files