Automated Compression
This directory contains example applications of various methods to automate the DNN compression process.
We will slowly release implementations of these methods.
FMReconstructionChannelPruner: add support for nn.Linear layers utils.py: add non_zero_channels() thinning: support removing channels from FC layers preceding Conv layers test_pruning.py: add test_row_pruning() scheduler: init from a dictionary of Maskers coach_if.py – fix imports of Clipped-PPO and TD3
Name | Last commit | Last update |
---|---|---|
.. | ||
amc | ||
README.md | ||
__init__.py | ||
plain20_cifar_baseline_training.yaml | ||
plain20_fine_tune.yaml | ||
vgg16_cifar_baseline_training.yaml |