Update train.py

Doc fix
2024-02-13 09:41:48 -08:00 · 2024-02-13 09:41:48 -08:00 · 1934760056
parent 0cec613c07
commit 1934760056
1 changed files with 1 additions and 1 deletions
--- 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