Per sample grad correctness util#532

Closed

psolikov wants to merge 38 commits intometa-pytorch:mainfrom

psolikov:per-sample-grad-correctness-util

Commits on Oct 25, 2022

support empty batches in memory manager and optimizer
Igor Shilov
committed
restore warning
Igor Shilov
committed
disable functorch test for 1.13+
Igor Shilov
committed

Commits on Oct 26, 2022

Merge branch 'main' of github.com:pytorch/opacus into ffuuugor_522
Igor Shilov
committed

Commits on Oct 27, 2022

0-batch tests
Igor Shilov
committed
lint
Igor Shilov
committed
EW test fix
Igor Shilov
committed
docstring up
Igor Shilov
committed
Implement per sample grads util and refactor code
Pavel Solikov
committed
Merge branch 'pytorch:main' into per-sample-grad-correctness-util
psolikov
authored

Commits on Oct 28, 2022

Add docs and refactor
Pavel Solikov
committed
Apply code style fixes
Pavel Solikov
committed
Merge branch 'main' into per-sample-grad-correctness-util
psolikov
authored
Fix flake8 errors
Pavel Solikov
committed

Commits on Oct 31, 2022

Implement per sample grads util and refactor code
Pavel Solikov
committed
Fixed issue with missing argument in MNIST example (#520)

Goutham Rajendran
authored and
Pavel Solikov
committed
Add docs and refactor
Pavel Solikov
committed
Apply code style fixes
Pavel Solikov
committed
Functorch gradients: investigation and fix (#510)

Igor Shilov
authored and
Pavel Solikov
committed
Fix flake8 errors
Pavel Solikov
committed
Add type hints
Pavel Solikov
committed
Refactor
Pavel Solikov
committed
Update docstrings
Pavel Solikov
committed
Fix reduction modes for EW
Pavel Solikov
committed

Commits on Nov 1, 2022

Rebase on #530, separate utils tests, refactor
Pavel Solikov
committed
Optimize imports
Pavel Solikov
committed
Merge remote-tracking branch 'origin/per-sample-grad-correctness-util' into per-sample-grad-correctness-util
Pavel Solikov
committed
Fix test
Pavel Solikov
committed
Add utility description to tutorial
Pavel Solikov
committed

Commits on Nov 7, 2022

Fix grad samples test
Pavel Solikov
committed
Fixed isort warnings
Pavel Solikov
committed
Fix grad samples zero batch test
Pavel Solikov
committed
Skip functorch test when unavailable
Pavel Solikov
committed

Commits on Nov 8, 2022

Merge branch 'main' into per-sample-grad-correctness-util
Pavel Solikov
committed
Fix merge
Pavel Solikov
committed
Isort fix
Pavel Solikov
committed
Fix docstring
Pavel Solikov
committed

Commits on Nov 17, 2022

Merge branch 'main' into per-sample-grad-correctness-util
psolikov
authored