remove irrelevant docs in optimizer_step (#4964)

Co-authored-by: chaton <thomas@grid.ai>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
This commit is contained in:
Rohit Gupta 2020-12-08 14:27:14 +05:30 committed by GitHub
parent 2393474350
commit cb28f79c50
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 4 deletions

View File

@ -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