device = torch.device("cuda")
real_batch = next(iter(dataloader))
plt.figure(figsize=(8, 8))
plt.axis("off")
plt.imshow(
np.transpose(
vutils.make_grid(real_batch[0].to(device)[:64],
padding=2,
normalize=True).cpu(), (1, 2,0)))
Created
February 13, 2020 06:43
-
-
Save SubhadityaMukherjee/e77b7b9cc80ee22c77a4c9df80c34efb to your computer and use it in GitHub Desktop.
load
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment