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

..

data/horovod

RC & Docs/changelog (#1776)

2020-05-11 21:57:53 -04:00

__init__.py

CI: split tests-examples (#990)

2020-03-25 07:46:27 -04:00

test_amp.py

Tests: refactor cleanup (#1744)

2020-05-10 13:15:28 -04:00

test_cpu.py

Replaces ddp .spawn with subprocess (#2029)

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

test_gpu.py

Replaces ddp .spawn with subprocess (#2029)

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

test_grad_norm.py

Mistake in parameters' grad norm tracking (#2012)

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

test_hooks.py

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

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

test_horovod.py

replace Hparams by init args (#1896)

2020-05-24 18:59:08 -04:00

test_hparams.py

hotfix to unblock hparams and OmniConf - removes auto_register_init_args by default (#2025)

2020-05-31 08:29:51 -04:00

test_restore.py

Replaces ddp .spawn with subprocess (#2029)

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