lightning/tests/trainer/optimization
Carlos Mocholí b1ed1db089
Keep global step update in the loop (#8856)
2021-09-14 19:21:39 +05:30
..
__init__.py ref: decouple apex second attemp part 6/n (#4060) 2020-10-10 15:28:25 -04:00
test_backward_calls.py Keep global step update in the loop (#8856) 2021-09-14 19:21:39 +05:30
test_manual_optimization.py Run plugin closure before `on_before_optimizer_step` [1/2] (#9288) 2021-09-07 11:52:20 +00:00
test_multiple_optimizers.py Enforce that the optimizer closure is executed when `optimizer_step` is overridden (#9360) 2021-09-08 12:24:57 +02:00
test_optimizers.py Add `init_optimizers` test for predict (#9440) 2021-09-13 09:05:04 +01:00