pytorch/test/optim
Michael Lazos b193f295b6 Add capturable ASGD impl (#107857)
Add capturable ASGD impl + test

Pull Request resolved: https://github.com/pytorch/pytorch/pull/107857
Approved by: https://github.com/janeyx99
2023-09-07 06:30:30 +00:00
..
test_lrscheduler.py [BE] fix type iteration typo in test_lrscheduler.py (#106908) 2023-08-09 23:56:06 +00:00
test_optim.py Add capturable ASGD impl (#107857) 2023-09-07 06:30:30 +00:00
test_swa_utils.py Dont run test files that are already run in test_optim (#103017) 2023-06-06 17:31:21 +00:00