lightning/tests/trainer/optimization
Carlos Mocholí b7920b1c84 Fix logging on_train_batch_end in a callback with multiple optimizers (#5521)
* Start with the failing test

* Then fix the failing test

* Update CHANGELOG
2021-02-03 19:41:46 +01:00
..
__init__.py ref: decouple apex second attemp part 6/n (#4060) 2020-10-10 15:28:25 -04:00
test_backward_calls.py Fix bug comparing max_steps to global step which inits at 0 (#4278) 2020-10-22 13:58:59 +01:00
test_manual_optimization.py Refactor setup_training and remove test_mode (#5388) 2021-02-03 19:41:42 +01:00
test_multiple_optimizers.py Fix logging on_train_batch_end in a callback with multiple optimizers (#5521) 2021-02-03 19:41:46 +01:00
test_optimizers.py deprecate enable_pl_optimizer as it is not restored properly (#5244) 2021-01-26 14:29:46 +01:00