Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update docs about transfer_batch_to_device hook while using DP (#7343) 

* update

* clarify

* update
This commit is contained in:
Adrian Wälchli 2021-05-05 15:51:16 +02:00 committed by GitHub
parent 626ef08694
commit 49c8740338
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
docs/source/advanced/multi_gpu.rst
View File

@ -288,9 +288,13 @@ after which the root node will aggregate the results.
.. warning:: DP use is discouraged by PyTorch and Lightning. State is not maintained on the replicas created by the
:class:`~torch.nn.DataParallel` wrapper and you may see errors or misbehavior if you assign state to the module
in the ``forward()`` or ``*_step()`` methods. For the same reason we do cannot fully support
in the ``forward()`` or ``*_step()`` methods. For the same reason we cannot fully support
:ref:`manual_optimization` with DP. Use DDP which is more stable and at least 3x faster.
.. warning:: DP only supports scattering and gathering primitive collections of tensors like lists, dicts, etc.
Therefore the :meth:`~pytorch_lightning.core.hooks.ModelHooks.transfer_batch_to_device` hook does not apply in
this mode and if you have overridden it, it will not be called.
.. testcode::
:skipif: torch.cuda.device_count() < 2