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.
zero_grad
1 parent a787097 commit 16392a7Copy full SHA for 16392a7
docs/source/guides/speed.rst
@@ -450,7 +450,7 @@ Set Grads to None
450
In order to modestly improve performance, you can override :meth:`~pytorch_lightning.core.lightning.LightningModule.optimizer_zero_grad`.
451
452
For a more detailed explanation of pros / cons of this technique,
453
-read `this <https://pytorch.org/docs/master/optim.html#torch.optim.Optimizer.zero_grad>`_ documentation by the PyTorch team.
+read the documentation for :meth:`~torch.optim.Optimizer.zero_grad` by the PyTorch team.
454
455
.. testcode::
456
0 commit comments