Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
lightning/tests/callbacks
History
chaton 02152c1729
Simplify optimization Logic (#4984) 
* Rely on ddp plugin for blocking sync behaviour, and skip if we're using manual optimization

* debug

* Revert "debug"

This reverts commit ccca6b6b

* Expose manual reduce for automatic optimization

* Add input arguments

* Enable parity test

* clean imports

* Expose hook after to ensure we reset

* Fix naming

* add

* fix test

* uniformize optimizer logic

* resolve test

* resovle flake8

* resolve amp bug

* update tests

* remove bug

* remove optimizer_step in accelerators

* typo

* update lightning optimizer

* set doesn't work with ddp_spawn

* resolve flake8

* update threshold

* ignore pyright

* correct codeFactor

* remove useless if

* remove zer_grad function

* simplify step

* remove typo

* resolve bug

* Apply suggestions from code review

* update on comments

* resolve bugs

* remove tests

* Update pytorch_lightning/trainer/configuration_validator.py

Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>

* simplify testing

* add more tests

Co-authored-by: SeanNaren <sean@grid.ai>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
 		2020-12-07 12:55:49 +00:00
..
__init__.py
…
test_callback_hook_outputs.py notices (#4118) 2020-10-13 07:18:07 -04:00
test_callbacks.py Simplify optimization Logic (#4984) 2020-12-07 12:55:49 +00:00
test_early_stopping.py [TEST] Min steps override early stopping (#4283) 2020-12-04 17:10:14 +01:00
test_gpu_stats_monitor.py limit monitor callback with log_every_n_steps (#3881) 2020-10-22 16:38:03 +05:30
test_lr_monitor.py add option to log momentum (#4384) 2020-10-28 21:56:58 +05:30
test_progress_bar.py fix incomplete progress bar when refresh_rate > num batches (#4577) 2020-11-24 00:01:33 +01:00