Note: This repo doesn't work. Lost interest, maybe one day I'll attempt to finish it.
Deep Compression with Reinforcement Learning. arxiv This is a PyTorch implementation of the DECORE compression algorithm.
TODO:
- Improve the wrapper over the custom structured pruning flow in PyTorch.
- Add REINFORCE-based RL agents before CNN layers that dictate which channels to prune.