From ea76ad2b28b10bab9d2146669d5cda3bd642a8a3 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Tue, 13 Aug 2019 06:39:10 -0400 Subject: [PATCH] Update RequiredTrainerInterface.md --- docs/LightningModule/RequiredTrainerInterface.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/LightningModule/RequiredTrainerInterface.md b/docs/LightningModule/RequiredTrainerInterface.md index 34a077f924..e5a0722efc 100644 --- a/docs/LightningModule/RequiredTrainerInterface.md +++ b/docs/LightningModule/RequiredTrainerInterface.md @@ -52,7 +52,7 @@ class CoolModel(pl.LightningModule): y_hat = self.forward(x) return {'loss': F.cross_entropy(y_hat, y)(y_hat, y)} - def validation_step(self, batch, batch_nb): + def validation_step(self, batch, batch_nb, dataloader_i): # OPTIONAL x, y = batch y_hat = self.forward(x)