Skip to content

Instantly share code, notes, and snippets.

@rafalpronko rafalpronko/image_show_2 Secret
Created Feb 11, 2019

Embed
What would you like to do?
# 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
You can’t perform that action at this time.