downloading images from colab, note doesn't work on its own

downloading_images_from_colab.py

from google.colab import drive
drive.mount('./data')

from google.colab import files

###

# plot 6-image samples, save to file
f, axarr = plt.subplots(2,3, figsize=(16,8))
for i in range(2):
  for j in range(3):
    img = G.forward(generate_random(100)).view(64,64).detach().cpu().numpy()
    img = (img + 1.0)/2.0
    axarr[i,j].imshow(img, interpolation='none', cmap='gray')
    pass
  pass

t = datetime.datetime.fromtimestamp(time.time())
fname = "img_%s.png" % (t.strftime('%H-%M-%S'))
plt.savefig(fname)
files.download(fname)
plt.close(f)