lightning/tests/tests_pytorch/trainer/optimization
Carlos Mocholí f17cacb04b Remove nivida/apex (#16149) 2023-01-19 18:39:36 +01:00
..
__init__.py
test_backward_calls.py
test_manual_optimization.py Remove nivida/apex (#16149) 2023-01-19 18:39:36 +01:00
test_multiple_optimizers.py Use `torch.testing.assert_close` everywhere (#15031) 2022-10-07 16:59:04 +02:00
test_optimizers.py Do not warn about the scheduler's interval key during manual optim (#16308) 2023-01-12 14:30:31 +01:00