wassname/pytorch-lightning
1
0 Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-10-28 13:20:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

typing error

Browse Source 
it should be val_batch here.
This commit is contained in:
Caldera authored and GitHub committed 2020-04-14 00:16:54 +08:00
1 parent 3f1e4b953f
commit 1b45ddcd17
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/core/lightning.py
+1 -1
View File
@@ -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)