Skip to content

declearn v2.2.2

Released: 06/09/2023

This is a subminor release that patches a regression introduced in v2.2.0 but was undetected before.

Hotfix: fix DP-SGD with TorchModel

Since version 2.2.0, the functorch-based backend for computing and clipping sample-wise gradients of a TorchModel has attempted to benefit from the experimental functorch.compile API. However, this change, which has been tested to yield performance gains on fixed-size batches of inputs, turns out not to be compatible with variable-size batches - which are mandatory as part of DP-SGD due to the use of Poisson sampling.

As a consequence, this version drops the use of functorch.compile, restoring DP-SGD features with `TorchModel``.