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