This commit is contained in:
Shunta Komatsu 2020-03-27 21:43:12 +09:00 committed by GitHub
parent 3a93aaf9e2
commit ac6692d3e4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -307,7 +307,7 @@ class LightningModule(ABC, GradInformation, ModelIO, ModelHooks):
val_outs = [] val_outs = []
for val_batch in val_data: for val_batch in val_data:
out = validation_step(train_batch) out = validation_step(train_batch)
val_outs.append(out val_outs.append(out)
validation_epoch_end(val_outs) validation_epoch_end(val_outs)
Args: Args: