Skip to content

v2.1.0: ALDS (NeurIPS 2021) release and updates
Compare
Choose a tag to compare
@lucaslie lucaslie released this 12 Oct 15:38
· 2 commits to main since this release
v2.1.0
0932cbc

The new release contains code for a new paper with accompanying code, comparison methods, models, and datasets.

In addition to the previous papers that were covered by this codebase (PFP, SiPP, Lost), we also extended the repository to include our latest paper on pruning, which will be presented at NeurIPS 2021:

Compressing Neural Networks: Towards Determining the Optimal Layer-wise Decomposition

Check out the READMEs for more info.

Detailed release update:

  • ALDS algorithm in torchprune.
  • Various tensor decomposition methods as comparisons for ALDS.
  • More network and dataset support, including Glue Benchmark and huggingface transformers.
  • Experiment code, visualization, and paper reproducibility for ALDS.
Assets 2
Loading