Skip to content

Commit 729fd60

Browse files
author
Seppo Enarvi
committed
Fixed a reference in a docstring
1 parent f3529f4 commit 729fd60

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/lightning/pytorch/callbacks/weight_averaging.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -57,8 +57,7 @@ class WeightAveraging(Callback):
5757
provided by Lightning.
5858
5959
Note:
60-
To ensure that the :class:`AveragedModel` will contain all layers,
61-
:meth:`~lightning.pytorch.callbacks.weight_averaging.WeightAveraging.setup` will call
60+
To ensure that the :class:`AveragedModel` will contain all layers, ``setup()`` will call
6261
:meth:`~lightning.pytorch.core.hooks.ModelHooks.configure_model` before instantiating the
6362
:class:`AveragedModel`. However, that hook is not called in a strategy aware context, sharded models do not work
6463
with weight averaging, and a warning will be issued.

0 commit comments

Comments
 (0)