lightning/pytorch_lightning
Adrian Wälchli 8211256c46
data transfer model hook (+ refactor) (#1756)
* refactor and added hook


variant a


variant b


add test


revert rename


add changelog


docs

* resolve merge duplication

* overridden typo

* fix test

* tpu id

* raise if TPU not available

* re-use apply_to_collection function for parsing collections

* comment

* make utility function available to user

* documentation

* move changelog entry to top

* fix tpu transfer call

* fix call

* remove hardcoded string

* improve test

* call model hook by default

* Apply suggestions from code review

* rename utility function

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
2020-06-02 21:45:19 -04:00
..
callbacks Revert "Fixes EarlyStopping With Precision=16 (#1996)" (#2032) 2020-05-31 15:20:18 -04:00
core data transfer model hook (+ refactor) (#1756) 2020-06-02 21:45:19 -04:00
loggers fix(wandb): use same logger on multiple training loops (#2055) 2020-06-02 18:46:02 -04:00
logging Join Horovod workers at the end of trainer.fit() to prevent race conditions following training (#1786) 2020-05-12 09:15:25 +00:00
metrics New metric classes (#1326) (#1877) 2020-05-19 11:05:07 -04:00
overrides fix default arg (#1927) 2020-05-26 19:04:42 -04:00
profiler replace Hparams by init args (#1896) 2020-05-24 18:59:08 -04:00
pt_overrides Join Horovod workers at the end of trainer.fit() to prevent race conditions following training (#1786) 2020-05-12 09:15:25 +00:00
root_module Join Horovod workers at the end of trainer.fit() to prevent race conditions following training (#1786) 2020-05-12 09:15:25 +00:00
trainer data transfer model hook (+ refactor) (#1756) 2020-06-02 21:45:19 -04:00
utilities data transfer model hook (+ refactor) (#1756) 2020-06-02 21:45:19 -04:00
__init__.py data transfer model hook (+ refactor) (#1756) 2020-06-02 21:45:19 -04:00