pytorch/torch/testing
Rohan Varma c11412b4a8 [DDP] Support optim in backward after DDP init (#105995)
This allows in backward optimizers to be configured after DDP init, in
addition to before as was previously supported.

Differential Revision: [D47783347](https://our.internmc.facebook.com/intern/diff/D47783347/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/105995
Approved by: https://github.com/fegin
2023-07-29 01:36:25 +00:00
..
_internal [DDP] Support optim in backward after DDP init (#105995) 2023-07-29 01:36:25 +00:00
__init__.py document torch.testing.assert_allclose (#89526) 2022-12-01 11:22:50 +00:00
_comparison.py [BE] Enable ruff's UP rules and autoformat testing/ (#105425) 2023-07-18 21:04:39 +00:00
_creation.py disallow invalid value ranges in torch.testing.make_tensor (#96334) 2023-03-24 23:55:17 +00:00