Adrian WälchliandJirka Borovec 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

..

__init__.py

Improved docs for pytorch_lightning.core (continued) (#1483)

2020-04-16 12:04:55 -04:00

decorators.py

add rank warning (#1428)

2020-04-09 14:05:46 -04:00

grads.py

Mistake in parameters' grad norm tracking (#2012)

2020-06-02 18:51:09 -04:00

hooks.py

data transfer model hook (+ refactor) (#1756)

2020-06-02 21:45:19 -04:00

lightning.py

Replaces ddp .spawn with subprocess (#2029)

2020-06-01 11:00:32 -04:00

memory.py

Improved docs for pytorch_lightning.core (continued) (#1483)

2020-04-16 12:04:55 -04:00

model_saving.py

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.py

Join Horovod workers at the end of trainer.fit() to prevent race conditions following training (#1786)

2020-05-12 09:15:25 +00:00

saving.py

fix default arg (#1927)

2020-05-26 19:04:42 -04:00