From 0cec613c07a5ccaffc43250236b77c8ec1543be4 Mon Sep 17 00:00:00 2001 From: Karthik Venkataramani Date: Tue, 13 Feb 2024 09:40:57 -0800 Subject: [PATCH] Update introduction.rst Doc fix --- docs/source-pytorch/starter/introduction.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source-pytorch/starter/introduction.rst b/docs/source-pytorch/starter/introduction.rst index 3cfe872613..2c8d7d3f6b 100644 --- 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