Update gan.py

2019-08-14 08:43:50 -04:00 · 2019-08-14 08:43:50 -04:00 · c9117f74b2
parent 13f2d1ab1c
commit c9117f74b2
1 changed files with 0 additions and 3 deletions
--- 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)