Examples: using new API (#1056)

* using new API

* typo
This commit is contained in:
William Falcon
2020-03-05 19:31:57 -05:00
committed by GitHub
parent bb7356bcaa
commit 0ebfb78570
9 changed files with 16 additions and 17 deletions
@@ -143,7 +143,7 @@ class LightningTemplateModel(LightningModule):
# can also return just a scalar instead of a dict (return loss_val)
return output
def validation_end(self, outputs):
def validation_epoch_end(self, outputs):
"""
Called at the end of validation to aggregate outputs
:param outputs: list of individual outputs of each validation step