From 2ec8d61e94722f7ecc97e1add72f4ac693d2f612 Mon Sep 17 00:00:00 2001 From: weipengOO98 <63845580+weipengOO98@users.noreply.github.com> Date: Thu, 30 Apr 2020 19:58:42 +0800 Subject: [PATCH] Update new-project.rst (#1655) fix a typo --- docs/source/new-project.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/new-project.rst b/docs/source/new-project.rst index 7d81ba44a3..e3f3a892d9 100644 --- a/docs/source/new-project.rst +++ b/docs/source/new-project.rst @@ -100,7 +100,7 @@ To also add a validation loop add the following functions def validation_epoch_end(self, outputs): avg_loss = torch.stack([x['val_loss'] for x in outputs]).mean() tensorboard_logs = {'val_loss': avg_loss} - return {'val_loss': avg_loss, 'log': tensorboard_logs + return {'val_loss': avg_loss, 'log': tensorboard_logs} def val_dataloader(self): # TODO: do a real train/val split