Skip to content
This repository has been archived by the owner on May 1, 2023. It is now read-only.

PyTorch 0.4 support and new features

Pre-release
Pre-release
Compare
Choose a tag to compare
@nzmora nzmora released this 25 Jun 12:12
· 575 commits to master since this release
  • PyTorch 0.4 support
  • An implementation of Baidu's RNN pruning paper from ICLR 2017
    Narang, Sharan & Diamos, Gregory & Sengupta, Shubho & Elsen, Erich. (2017).
    Exploring Sparsity in Recurrent Neural Networks. (https://arxiv.org/abs/1704.05119)
  • Add a word language model pruning example using AGP and Baidu RNN pruning
  • Quantization aware training (4-bit quantization)
  • New models: pre-activation ResNet for ImageNet and CIFAR, and AlexNet with batch-norm
  • New quantization documentation content