benchmark
bb3a8be2 - Add pure-functorch variant of the dp-cifar10 model (#1178)

Commit
3 years ago
Add pure-functorch variant of the dp-cifar10 model (#1178) Summary: Pull Request resolved: https://github.com/pytorch/benchmark/pull/1178 Test Plan: - python test.py -k "test_functorch_dp_cifar10_train_cuda" - I'm not sure what else needs to be tested, taking suggestions Reviewed By: xuzhao9 Differential Revision: D39670612 Pulled By: zou3519 fbshipit-source-id: 58e8b430d2df4f52a0c9ad915051ad89c0374f00
Author
Parents
Loading