f79a13e495
* 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> |
||
---|---|---|
.. | ||
__init__.py | ||
test_callback_hook_outputs.py | ||
test_callbacks.py | ||
test_early_stopping.py | ||
test_finetuning_callback.py | ||
test_gpu_stats_monitor.py | ||
test_lambda_function.py | ||
test_lr_monitor.py | ||
test_progress_bar.py | ||
test_pruning.py | ||
test_quantization.py | ||
test_stochastic_weight_avg.py |