Skip to content

v0.2.1

Choose a tag to compare

@lxuechen lxuechen released this 18 Jul 02:33
· 25 commits to main since this release
  • The codebase now supports the following additional models

  • Minor rewrite of the privacy engine, sharing code between ghost and default clipping, thus making future maintaining potentially easier.

  • Add example file for fine-tuning of vision Transformers on CIFAR-10.

  • Use callback style implementation for playing with params and gradients after clipping.

  • Allow forward and backward hooks to use additional args and kwargs (not necessarily torch.Tensors).

  • Rewrite privacy accounting and remove GDP + CLT, since it under accounts.