Skip to content

Instantly share code, notes, and snippets.

@rafalpronko

rafalpronko/image_show_2 Secret

Created February 11, 2019 05:50
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save rafalpronko/e2770cc3342ceb1f46f952cd062dde41 to your computer and use it in GitHub Desktop.
Save rafalpronko/e2770cc3342ceb1f46f952cd062dde41 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
image_show_2
# create a data loader - I will use it for the training
train_loader = torch.utils.data.DataLoader(dataset=train_dataset_py,
batch_size=32,
shuffle=False)
test_loader = torch.utils.data.DataLoader(dataset=test_dataset_py,
batch_size=32,
shuffle=False)
data_iter = iter(train_loader)
images, label = data_iter.next()
imshowPytorch(torchvision.utils.make_grid(images[0]))
print(label[0])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment