diff --git a/README.md b/README.md index 08529f2b0c..4b78059a11 100644 --- a/README.md +++ b/README.md @@ -177,7 +177,7 @@ class LitAutoEncoder(pl.LightningModule): return embedding def training_step(self, batch, batch_idx): - # training_step defined the train loop. It is independent of forward + # training_step defines the train loop. It is independent of forward x, y = batch x = x.view(x.size(0), -1) z = self.encoder(x)