lightning/pytorch_lightning/root_module
Travis Addair acab068c74
Join Horovod workers at the end of trainer.fit() to prevent race conditions following training (#1786)
* Join Horovod workers at the end of trainer.fit() to prevent race conditions following training

* flake8

* flake8

Co-authored-by: Jirka <jirka.borovec@seznam.cz>
2020-05-12 09:15:25 +00:00
..
__init__.py add rank warning (#1428) 2020-04-09 14:05:46 -04:00
decorators.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
grads.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
hooks.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
memory.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
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