lightning/tests/tests_pytorch/trainer/optimization
Quazi Marufur Rahman 2577285dd5
Use LRScheduler for torch >= 1.14 otherwise use _LRScheduler (#15768)
Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com>
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
2022-12-12 13:33:26 +00:00
..
__init__.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_backward_calls.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_manual_optimization.py Use LRScheduler for torch >= 1.14 otherwise use _LRScheduler (#15768) 2022-12-12 13:33:26 +00:00
test_multiple_optimizers.py Use `torch.testing.assert_close` everywhere (#15031) 2022-10-07 16:59:04 +02:00
test_optimizers.py Use LRScheduler for torch >= 1.14 otherwise use _LRScheduler (#15768) 2022-12-12 13:33:26 +00:00