typing error

it should be val_batch here.
2026-09-09 11:32:07 +08:00 · 2020-04-14 00:16:54 +08:00
parent 3f1e4b953f
commit 1b45ddcd17
1 changed files with 1 additions and 1 deletions
@@ -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)