fix typo in Optimization (#5228)
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
(cherry picked from commit 5651c9cc1a
)
This commit is contained in:
parent
d19d711dcf
commit
edf1defb4e
|
@ -67,7 +67,7 @@ Under the hood Lightning does the following:
|
|||
.. code-block:: python
|
||||
|
||||
for epoch in epochs:
|
||||
for batch id data:
|
||||
for batch in data:
|
||||
loss = model.training_step(batch, batch_idx, ...)
|
||||
loss.backward()
|
||||
optimizer.step()
|
||||
|
|
Loading…
Reference in New Issue