Training a DCGAN in PyTorch