From cb28f79c50814f29b5a0145d10e1664345787049 Mon Sep 17 00:00:00 2001 From: Rohit Gupta Date: Tue, 8 Dec 2020 14:27:14 +0530 Subject: [PATCH] remove irrelevant docs in optimizer_step (#4964) Co-authored-by: chaton Co-authored-by: Jirka Borovec --- pytorch_lightning/core/lightning.py | 4 ---- 1 file changed, 4 deletions(-) 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