Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update README.md 

Doc fix
This commit is contained in:
Karthik Venkataramani 2024-02-13 09:39:44 -08:00 committed by GitHub
parent 183a2a9e33
commit fda3fbf32d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -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