lightning/tests/trainer/optimization
Carlos Mocholí 3f355d0eb7
Remove manual tracking of optimizer steps (#9957)
2021-10-18 12:43:06 +00:00
..
__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 Remove manual tracking of optimizer steps (#9957) 2021-10-18 12:43:06 +00:00
test_multiple_optimizers.py Add `enable_model_summary` flag and deprecate `weights_summary` (#9699) 2021-10-13 17:20:54 +05:30
test_optimizers.py (1/n) tests: Use strategy flag instead of accelerator for training strategies (#9931) 2021-10-16 20:40:25 +05:30