From 7bb245d17975697658afae0b498f309c4fc439a6 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 5 Mar 2020 20:11:27 -0500 Subject: [PATCH] Update README.md --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index f1edb3b540..c254b2009c 100644 --- a/README.md +++ b/README.md @@ -145,7 +145,7 @@ To use lightning do 2 things: y_hat = self.forward(x) return {'val_loss': F.cross_entropy(y_hat, y)} - def validation_epoch_end(self, outputs): + def validation_end(self, outputs): # OPTIONAL avg_loss = torch.stack([x['val_loss'] for x in outputs]).mean() tensorboard_logs = {'val_loss': avg_loss} @@ -157,7 +157,7 @@ To use lightning do 2 things: y_hat = self.forward(x) return {'test_loss': F.cross_entropy(y_hat, y)} - def test_epoch_end(self, outputs): + def test_end(self, outputs): # OPTIONAL avg_loss = torch.stack([x['test_loss'] for x in outputs]).mean() tensorboard_logs = {'test_loss': avg_loss}