From fda3fbf32d11cb012e068f0420b23360e5ecac0d Mon Sep 17 00:00:00 2001
From: Karthik Venkataramani <karthik0@vt.edu>
Date: Tue, 13 Feb 2024 09:39:44 -0800
Subject: [PATCH] Update README.md

Doc fix
---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index c27e2a3b79..2bbd105ecd 100644
--- a/README.md
+++ b/README.md
@@ -148,7 +148,7 @@ class LitAutoEncoder(L.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