Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix enumerate usage (#12949)

This commit is contained in:
Na Young Jun 2022-05-02 06:29:56 -07:00 committed by GitHub
parent 0f418a8640
commit 0e4c4424fd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/source/model/train_model_basic.rst
View File

@ -119,7 +119,7 @@ Under the hood, the Lightning Trainer runs the following training loop on your b
autoencoder = LitAutoEncoder(encoder, decoder) autoencoder = LitAutoEncoder(encoder, decoder)
optimizer = autoencoder.configure_optimizers() optimizer = autoencoder.configure_optimizers()
for batch, batch_idx in enumerate(train_loader): for batch_idx, batch in enumerate(train_loader):
loss = autoencoder(batch, batch_idx) loss = autoencoder(batch, batch_idx)
loss.backward() loss.backward()