lightning/tests/callbacks
Kaushik B f79a13e495
[Model Parallel] Add configure sharded model hook (#6679)
* Add base hook for model parallel

* fix callback signature

* Simplify hook

* Add hook logic

* add tests

* add property setter

* add logic for being called once

* Update changelog

* Fix

* fix return type

* fix lambda callback test

* Fix tests

* Apply code suggestions

* add logic for setup_optimizers_predispatch

* add common dummy model

* Swap call order

* Remove test that isn't needed anymore

* Update tests

* Add a bit more doc

* Few code review fixes

* Update pytorch_lightning/accelerators/accelerator.py

Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>

* Change hook name

* Fix test

* Test setup hook, refactor names

* Swap call order of callbacks and model initialization

* Change name of context manager

Co-authored-by: SeanNaren <sean@grid.ai>
Co-authored-by: Sean Naren <sean.narenthiran@gmail.com>
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
2021-03-29 14:50:51 -06:00
..
__init__.py Progress bar callback (#1450) 2020-04-23 20:46:18 -04:00
test_callback_hook_outputs.py Add outputs param for `on_val/test_epoch_end` hooks (#6120) 2021-03-16 12:15:16 -04:00
test_callbacks.py [Model Parallel] Add configure sharded model hook (#6679) 2021-03-29 14:50:51 -06:00
test_early_stopping.py Do not add return dict items to callback_metrics (#6682) 2021-03-26 14:05:20 +01:00
test_finetuning_callback.py [accelerator][FeatBugFix] Improve manual optimization API (#5771) 2021-02-16 16:00:35 -05:00
test_gpu_stats_monitor.py Refactor: skipif for Windows 2/n (#6268) 2021-03-02 09:36:01 +00:00
test_lambda_function.py Refactor simplify tests (#5861) 2021-02-08 11:52:02 +01:00
test_lr_monitor.py Typing for tests 1/n (#6313) 2021-03-09 11:27:15 +00:00
test_progress_bar.py Add Trainer.validate(…) method to run one validation epoch (#4948) 2021-03-11 03:46:37 +01:00
test_pruning.py Typing for tests 1/n (#6313) 2021-03-09 11:27:15 +00:00
test_quantization.py Typing for tests 1/n (#6313) 2021-03-09 11:27:15 +00:00
test_stochastic_weight_avg.py Typing for tests 1/n (#6313) 2021-03-09 11:27:15 +00:00