From bdd86087e6ad08c2a057187d094a93fb958a3653 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Fri, 16 Aug 2019 10:07:56 -0400 Subject: [PATCH] updated docs --- docs/LightningModule/RequiredTrainerInterface.md | 1 + 1 file changed, 1 insertion(+) diff --git a/docs/LightningModule/RequiredTrainerInterface.md b/docs/LightningModule/RequiredTrainerInterface.md index 06c57dd5..01a76342 100644 --- a/docs/LightningModule/RequiredTrainerInterface.md +++ b/docs/LightningModule/RequiredTrainerInterface.md @@ -263,6 +263,7 @@ def validation_step(self, data_batch, batch_nb): loss = self.loss(out, x) # log 6 example images + # or generated text... or whatever sample_imgs = x[:6] grid = torchvision.utils.make_grid(sample_imgs) self.experiment.add_image('example_images', grid, 0)