mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-08-21 11:20:03 +08:00
Travis AddairandJirka
acab068c74
* 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>
12 lines
459 B
Python
12 lines
459 B
Python
"""
|
|
.. warning:: `root_module.memory` module has been renamed to `core.memory` since v0.6.0.
|
|
The deprecated module name will be removed in v0.8.0.
|
|
"""
|
|
|
|
from pytorch_lightning.utilities import rank_zero_warn
|
|
|
|
rank_zero_warn("`root_module.memory` module has been renamed to `core.memory` since v0.6.0."
|
|
" The deprecated module name will be removed in v0.8.0.", DeprecationWarning)
|
|
|
|
from pytorch_lightning.core.memory import * # noqa: F403 E402
|