This project consists of following files:
- common.py (Some common utils)
- metrics.py (Metrics for evaluating the importance of the weight elements)
- pruner.py (Class to prune the weights)
- reconstrctor.py (Class to transform the weights)
- train.py (Method for training the neural network)
Usage can be reffered to tutorial.ipynb
.