GAN example doesn't work
🐛 Bug
The GAN example for MNIST doesn't work as advised in the script basic_gan_module.py
Please reproduce using the BoringModel and post here
To reproduce the issue just run:
basic_gan_module.py --gpus 1
To Reproduce

Expected behavior
Train the GAN network well for MNIST dataset with the default setting.
Environment

CUDA:

GPU:

GeForce GTX 1080 Ti


available:         True
version:           10.1


Packages:

numpy:             1.19.1
pyTorch_debug:     False
pyTorch_version:   1.6.0
pytorch-lightning: 1.0.2
tqdm:              4.50.2


System:

OS:                Linux
architecture:

64bit
ELF


processor:         x86_64
python:            3.8.5
version:           #1 SMP Tue Aug 25 17:23:54 UTC 2020