Skip to content

Commit 07d689f

Browse files
committed
add a workaround to fix
1 parent baf2ed2 commit 07d689f

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

src/lightning/pytorch/trainer/trainer.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -563,15 +563,16 @@ def fit(
563563
recommend using ``weights_only=True``. For more information, please refer to the
564564
`PyTorch Developer Notes on Serialization Semantics <https://docs.pytorch.org/docs/main/notes/serialization.html#id3>`_.
565565
566+
For more information about multiple dataloaders, see this :ref:`section <multiple-dataloaders>`.
567+
568+
:rtype: :py:obj:`None`
569+
566570
Raises:
567571
TypeError:
568572
If ``model`` is not :class:`~lightning.pytorch.core.LightningModule` for torch version less than
569573
2.0.0 and if ``model`` is not :class:`~lightning.pytorch.core.LightningModule` or
570574
:class:`torch._dynamo.OptimizedModule` for torch versions greater than or equal to 2.0.0 .
571575
572-
.. note::
573-
For more information about multiple dataloaders, see this :ref:`section <multiple-dataloaders>`.
574-
575576
"""
576577
model = _maybe_unwrap_optimized(model)
577578
self.strategy._lightning_module = model

0 commit comments

Comments
 (0)