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 3aec635 commit 4f38156Copy full SHA for 4f38156
docs/source-pytorch/advanced/training_tricks.rst
@@ -73,6 +73,10 @@ procedure starts.
73
For a more detailed explanation of SWA and how it works, read
74
`this post <https://pytorch.org/blog/pytorch-1.6-now-includes-stochastic-weight-averaging>`__ by the PyTorch team.
75
76
+.. seealso::
77
+ The :class:`~lightning.pytorch.callbacks.WeightAveraging` callback and
78
+ :class:`~lightning.pytorch.callbacks.StochasticWeightAveraging` callback
79
+
80
.. testcode::
81
82
from lightning.pytorch.callbacks import StochasticWeightAveraging, WeightAveraging
0 commit comments