From 1b45ddcd17cb6b7de6293eec9cf9bc2f9158f718 Mon Sep 17 00:00:00 2001 From: Caldera Date: Tue, 14 Apr 2020 00:16:54 +0800 Subject: [PATCH] typing error it should be val_batch here. --- pytorch_lightning/core/lightning.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/core/lightning.py b/pytorch_lightning/core/lightning.py index a34a5a93..04a386d7 100644 --- a/pytorch_lightning/core/lightning.py +++ b/pytorch_lightning/core/lightning.py @@ -539,7 +539,7 @@ class LightningModule(ABC, GradInformation, ModelIO, ModelHooks): # the pseudocode for these calls val_outs = [] for val_batch in val_data: - out = validation_step(train_batch) + out = validation_step(val_batch) val_outs.append(out) validation_epoch_end(val_outs)