From c9117f74b25976cb246b2547e80f35db51862006 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 14 Aug 2019 08:43:50 -0400 Subject: [PATCH] Update gan.py --- examples/templates/gan.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/examples/templates/gan.py b/examples/templates/gan.py index e5e5ddb560..e8ffe86115 100644 --- a/examples/templates/gan.py +++ b/examples/templates/gan.py @@ -77,9 +77,6 @@ class GAN(pl.LightningModule): super(GAN, self).__init__() self.hparams = hparams - # let trainer show inputs/outputs for each layer (generator in this case) - # self.example_input_array = torch.rand(10, hparams.latent_dim) - # networks mnist_shape = (1, 28, 28) self.generator = Generator(latent_dim=hparams.latent_dim, img_shape=mnist_shape)