wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-12 12:40:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
* 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>
This commit is contained in:
Travis Addair
2020-05-12 09:15:25 +00:00
committed by GitHub
co-authored by Jirka
parent 7b60d49432
commit acab068c74
16 changed files with 34 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/root_module/model_saving.py
+2 -1
View File
@@ -4,7 +4,8 @@
"""
from pytorch_lightning.utilities import rank_zero_warn
from pytorch_lightning.core.saving import * # noqa: F403
rank_zero_warn("`root_module.model_saving` module has been renamed to `core.saving` since v0.6.0."
" The deprecated module name will be removed in v0.8.0.", DeprecationWarning)
from pytorch_lightning.core.saving import * # noqa: F403 E402