We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f3529f4 commit 729fd60Copy full SHA for 729fd60
src/lightning/pytorch/callbacks/weight_averaging.py
@@ -57,8 +57,7 @@ class WeightAveraging(Callback):
57
provided by Lightning.
58
59
Note:
60
- To ensure that the :class:`AveragedModel` will contain all layers,
61
- :meth:`~lightning.pytorch.callbacks.weight_averaging.WeightAveraging.setup` will call
+ To ensure that the :class:`AveragedModel` will contain all layers, ``setup()`` will call
62
:meth:`~lightning.pytorch.core.hooks.ModelHooks.configure_model` before instantiating the
63
:class:`AveragedModel`. However, that hook is not called in a strategy aware context, sharded models do not work
64
with weight averaging, and a warning will be issued.
0 commit comments