From 7898d0c02a0c39893f7bb51df363080e389f0d89 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Tue, 13 Aug 2019 11:51:31 -0400 Subject: [PATCH] updated optimizer_step docs --- docs/LightningModule/RequiredTrainerInterface.md | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/docs/LightningModule/RequiredTrainerInterface.md b/docs/LightningModule/RequiredTrainerInterface.md index e4d1c670..a5ca5bd2 100644 --- a/docs/LightningModule/RequiredTrainerInterface.md +++ b/docs/LightningModule/RequiredTrainerInterface.md @@ -200,7 +200,10 @@ def configure_optimizers(self): ### validation_step ``` {.python} -def validation_step(self, data_batch, batch_nb, dataloader_i) +def validation_step(self, data_batch, batch_nb) + +# if have multiple val dataloaders: +def validation_step(self, data_batch, batch_nb, dataloader_idx) ``` **OPTIONAL** If you don't need to validate you don't need to implement this method.