Lessone 12 - wgan - how to get individual pictures


In dl2/wgan.ipynb, the following code produces an 8x8 matrix > with the fake images:

netD.eval(); netG.eval();
fake = netG(fixed_noise).data.cpu()
faked = np.clip(md.trn_ds.denorm(fake),0,1)

plt.imshow(gallery(faked, 8));

How can we modify this code to get only one fake image with normal resolution ? I’m using a GAN for data augmentation so I cannot use the matrix.