Update README.md

2026-09-09 11:32:07 +08:00 · 2019-08-13 06:42:25 -04:00
parent 190a3a9260
commit 6f3152bcd6
1 changed files with 1 additions and 1 deletions
@@ -91,7 +91,7 @@ class CoolModel(pl.LightningModule):
        # OPTIONAL
        x, y = batch
        y_hat = self.forward(x)
-        return {'val_loss': self.my_loss(y_hat, y)}
+        return {'val_loss': F.cross_entropy(y_hat, y)}

    def validation_end(self, outputs):
        # OPTIONAL