lightning/tests/models
Travis Addair 7024177f7d
Added Horovod distributed backend (#1529)
* Initial commit of Horovod distributed backend implementation

* Update distrib_data_parallel.py

* Update distrib_data_parallel.py

* Update tests/models/test_horovod.py

Co-Authored-By: Jirka Borovec <Borda@users.noreply.github.com>

* Update tests/models/test_horovod.py

Co-Authored-By: Jirka Borovec <Borda@users.noreply.github.com>

* Fixed tests

* Added six

* tests

* Install tox for GitHub CI

* Retry tests

* Catch all exceptions

* Skip cache

* Remove tox

* Restore pip cache

* Remove the cache

* Restore pip cache

* Remove AMP

Co-authored-by: William Falcon <waf2107@columbia.edu>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: J. Borovec <jirka.borovec@seznam.cz>
2020-04-22 17:39:08 -04:00
..
data/horovod Added Horovod distributed backend (#1529) 2020-04-22 17:39:08 -04:00
__init__.py CI: split tests-examples (#990) 2020-03-25 07:46:27 -04:00
test_amp.py default test logger (#1478) 2020-04-21 20:33:10 -04:00
test_cpu.py default test logger (#1478) 2020-04-21 20:33:10 -04:00
test_gpu.py default test logger (#1478) 2020-04-21 20:33:10 -04:00
test_hooks.py Call on_before_zero_grad model hook (#1493) 2020-04-16 12:01:41 -04:00
test_horovod.py Added Horovod distributed backend (#1529) 2020-04-22 17:39:08 -04:00
test_restore.py default test logger (#1478) 2020-04-21 20:33:10 -04:00