Skip to content

v1.4.1

Compare
Choose a tag to compare
@VainF VainF released this 21 Jul 04:19
· 84 commits to master since this release

What's Changed

  • Add Isomorphic Pruning, an improved algorithm for global pruning.
  • Unify local/global/isomorphic pruning with Scope for importance ranking
  • Allow user-defined scope for importance ranking. The following key-value pair (model.layer1, model.layer2): 0.4 will perform global ranking only within layer1 and layer2, with the pruning ratio of 40%.
    pruner = tp.pruner.MetaPruner(
        ...
        global_pruning=True,
        pruning_ratio=0.5, # default pruning ratio
        pruning_ratio_dict = {(model.layer1, model.layer2): 0.4, model.layer3: 0.2}, 
        # Global pruning will be performed on layer1 and layer2
    )
  • Bugfixing

New Contributors

Full Changelog: v1.4.0...v1.4.1