Skip to content

Commit 4f38156

Browse files
senarviSkafteNicki
andauthored
Added seealso to WeightAveraging and StochasticWeightAveraging
Co-authored-by: Nicki Skafte Detlefsen <[email protected]>
1 parent 3aec635 commit 4f38156

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

docs/source-pytorch/advanced/training_tricks.rst

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -73,6 +73,10 @@ procedure starts.
7373
For a more detailed explanation of SWA and how it works, read
7474
`this post <https://pytorch.org/blog/pytorch-1.6-now-includes-stochastic-weight-averaging>`__ by the PyTorch team.
7575

76+
.. seealso::
77+
The :class:`~lightning.pytorch.callbacks.WeightAveraging` callback and
78+
:class:`~lightning.pytorch.callbacks.StochasticWeightAveraging` callback
79+
7680
.. testcode::
7781

7882
from lightning.pytorch.callbacks import StochasticWeightAveraging, WeightAveraging

0 commit comments

Comments
 (0)