lightning/pytorch_lightning/utilities
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
..
__init__.py data transfer model hook (+ refactor) (#1756) 2020-06-02 21:45:19 -04:00
apply_func.py data transfer model hook (+ refactor) (#1756) 2020-06-02 21:45:19 -04:00
device_dtype_mixin.py New metric classes (#1326) (#1877) 2020-05-19 11:05:07 -04:00
distributed.py model checkpint on rank_zero_only & global rank state (#1408) 2020-04-24 17:21:00 -04:00
exceptions.py Profiler summary (#1259) 2020-03-31 08:57:48 -04:00
memory.py Add toma comments to auto_scale_batch_size (#1994) 2020-05-29 05:57:50 +00:00
parsing.py ddp pickle 2020-04-27 06:51:42 -04:00