Update README.md

2026-09-09 11:32:07 +08:00 · 2019-08-13 06:37:30 -04:00
parent b1bf0a8d9b
commit 4f0cf1e970
1 changed files with 1 additions and 1 deletions
@@ -87,7 +87,7 @@ class CoolModel(pl.LightningModule):
        y_hat = self.forward(x)
        return {'loss': F.cross_entropy(y_hat, y)}

-    def validation_step(self, batch, batch_nb):
+    def validation_step(self, batch, batch_nb, dataset_i):
        # OPTIONAL
        x, y = batch
        y_hat = self.forward(x)