From 6e805501371c639a2c029ffad8006a706ca9cf96 Mon Sep 17 00:00:00 2001 From: rohitgr7 Date: Fri, 4 Dec 2020 02:50:29 +0530 Subject: [PATCH] remove irrelevant docs in optimizer_step --- 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 78fc740e389aa..8ce472a0e5dc8 100644 --- a/pytorch_lightning/core/lightning.py +++ b/pytorch_lightning/core/lightning.py @@ -1231,10 +1231,6 @@ def optimizer_step(self, epoch, batch_idx, optimizer, optimizer_idx, 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 on_tpu and TPU_AVAILABLE: xm.optimizer_step(optimizer, optimizer_args={'closure': optimizer_closure, **kwargs})