We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f9c9e39 commit 2ec8d61Copy full SHA for 2ec8d61
docs/source/new-project.rst
@@ -100,7 +100,7 @@ To also add a validation loop add the following functions
100
def validation_epoch_end(self, outputs):
101
avg_loss = torch.stack([x['val_loss'] for x in outputs]).mean()
102
tensorboard_logs = {'val_loss': avg_loss}
103
- return {'val_loss': avg_loss, 'log': tensorboard_logs
+ return {'val_loss': avg_loss, 'log': tensorboard_logs}
104
105
def val_dataloader(self):
106
# TODO: do a real train/val split
0 commit comments