lightning

Commit Graph

Author	SHA1	Message	Date
Carlos Mocholí	33066f8fd9	Add `on_predict_{batch,epoch}_{start,end}` and `Callback.on_predict_{start,end}` (#7141 ) * Update hooks typing and predict hooks * Update CHANGELOG * Progress * Progress * Add back `on_predict_{start,end}` * Typing and fix * Update tests/trainer/logging_/test_logger_connector.py * Update tests/callbacks/test_lambda_function.py	2021-04-22 10:05:28 -04:00
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
Jirka Borovec	79d42d83e7	formatting 3/n: PL modules (#5716 ) * cb * log * prof * tune * flake8	2021-02-08 14:28:38 -05:00
Wansoo Kim	61f415f2ac	Add LambdaCallback (#5347 ) * Add LambdaCallback * docs * add pr link # Conflicts: # CHANGELOG.md * convention * Fix Callback Typo * Update pytorch_lightning/callbacks/lambda_cb.py Co-authored-by: Nicki Skafte <skaftenicki@gmail.com> * Update pytorch_lightning/callbacks/lambda_cb.py Co-authored-by: Nicki Skafte <skaftenicki@gmail.com> * Update pytorch_lightning/callbacks/lambda_cb.py Co-authored-by: Nicki Skafte <skaftenicki@gmail.com> * use Misconfigureation * update docs * sort export * use inspect * string fill * use fast dev run * isort * remove unused import * sort * hilightning * highlighting * highlighting * remove debug log * eq * res * results * add misconfig exception test * use pytest raises * fix * Apply suggestions from code review Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com> * Update pytorch_lightning/callbacks/lambda_cb.py Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com> * hc * rm pt * fix * try fix * whitespace * new hook * add raise * fix * remove unused * rename Co-authored-by: Jirka Borovec <jirka.borovec@seznam.cz> Co-authored-by: Nicki Skafte <skaftenicki@gmail.com> Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com> Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com> Co-authored-by: chaton <thomas@grid.ai>	2021-01-13 04:42:49 -05:00

4 Commits