Skip to content

Opacus v0.15.0

Choose a tag to compare

@karthikprasad karthikprasad released this 25 Nov 01:26
· 513 commits to main since this release

New Features

  • DDP support for faster distributed training (#196)
  • Support of GRU and RNN. Refactored LSTM implementation. (#222)
  • PyTorch Lightning Demo (#244)

Bug fixes

  • Improve nn.Linear grad sampler memory consumption (#192)
  • Update Opacus to stop using deprecated torch.set_deterministic (#197)
  • Fix optimizer.step after engine.detach()
  • Test fixes

Miscellaneous

  • Better validation error reporting (#199)
  • grad sampler type checking (#241)