diff --git a/pytorch_lightning/core/lightning.py b/pytorch_lightning/core/lightning.py index f1a0c725e2..671084cb2f 100644 --- a/pytorch_lightning/core/lightning.py +++ b/pytorch_lightning/core/lightning.py @@ -1232,10 +1232,6 @@ class LightningModule( optimizer.step(closure=optimizer_closure) optimizer.zero_grad() - Note: - If you also override the :meth:`~pytorch_lightning.core.hooks.ModelHooks.on_before_zero_grad` - model hook don't forget to add the call to it before ``optimizer.zero_grad()`` yourself. - """ if not isinstance(optimizer, LightningOptimizer): # wraps into LightingOptimizer only for running step