lightning/tests/trainer/optimization
Carlos Mocholí 15d943089d
Enforce that the optimizer closure is executed when `optimizer_step` is overridden (#9360)
2021-09-08 12:24:57 +02:00
..
__init__.py ref: decouple apex second attemp part 6/n (#4060) 2020-10-10 15:28:25 -04:00
test_backward_calls.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
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 CI: precommit - docformatter (#8584) 2021-09-06 12:49:09 +00:00