fix bug

Author: StoneT2000

SP21/GAN/vanilla_gan.ipynb

@@ -148,7 +148,7 @@
     "            ### Train Generator ###\n",
     "            generator_optim.zero_grad()\n",
     "            \n",
-    "            latent_input = torch.randn((batch_size, 1, *latent_shape)).to(device)\n",
+    "            latent_input = torch.randn((len(real_images), 1, *latent_shape)).to(device)\n",
     "            fake_images = generator(latent_input)\n",
     "\n",
     "            fake_res = discriminator(fake_images)\n",