Update introduction.rst

Doc fix
2024-02-13 09:40:57 -08:00 · 2024-02-13 09:40:57 -08:00 · 0cec613c07
parent fda3fbf32d
commit 0cec613c07
1 changed files with 1 additions and 1 deletions
--- a/docs/source-pytorch/starter/introduction.rst
+++ b/docs/source-pytorch/starter/introduction.rst
@ -133,7 +133,7 @@ A LightningModule enables your PyTorch nn.Module to play together in complex way
            x = x.view(x.size(0), -1)
            z = self.encoder(x)
            x_hat = self.decoder(z)
-            loss = nn.functional.mse_loss(x_hat, x)
+            loss = nn.functional.mse_loss(x_hat, y)
            # Logging to TensorBoard (if installed) by default
            self.log("train_loss", loss)
            return loss