From 19347600564c0ab4fbda2e80263fc337f2297508 Mon Sep 17 00:00:00 2001 From: Karthik Venkataramani Date: Tue, 13 Feb 2024 09:41:48 -0800 Subject: [PATCH] Update train.py Doc fix --- docs/source-app/code_samples/convert_pl_to_app/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source-app/code_samples/convert_pl_to_app/train.py b/docs/source-app/code_samples/convert_pl_to_app/train.py index e9f9de3a6a..ea4a983c9d 100644 --- a/docs/source-app/code_samples/convert_pl_to_app/train.py +++ b/docs/source-app/code_samples/convert_pl_to_app/train.py @@ -29,7 +29,7 @@ class LitAutoEncoder(pl.LightningModule): x = x.view(x.size(0), -1) z = self.encoder(x) x_hat = self.decoder(z) - loss = F.mse_loss(x_hat, x) + loss = F.mse_loss(x_hat, y) self.log("train_loss", loss) return loss