lightning/tests/trainer/optimization
Carlos Mocholí 48b6292cf0
Move optimizer step and clipping into the `PrecisionPlugin` (#10143)
2021-10-26 17:26:26 +02:00
..
__init__.py
test_backward_calls.py Keep global step update in the loop (#8856) 2021-09-14 19:21:39 +05:30
test_manual_optimization.py Move optimizer step and clipping into the `PrecisionPlugin` (#10143) 2021-10-26 17:26:26 +02: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 Change default value of the `max_steps` Trainer argument from `None` to `-1` (#9460) 2021-10-25 20:21:33 +00:00