Skip to content

Instantly share code, notes, and snippets.

@vinimonteiro

vinimonteiro/nn_pytorch_data.py

Created December 17, 2021 14:54
Show Gist options
  • Save vinimonteiro/e3f1ea2e19490a44665b33b90804f13b to your computer and use it in GitHub Desktop.
Save vinimonteiro/e3f1ea2e19490a44665b33b90804f13b to your computer and use it in GitHub Desktop.
Download ZIP
nn_pytorch_data
Raw
nn_pytorch_data.py
train = datasets.MNIST("", train=True, download=True,
transform = transforms.Compose([transforms.ToTensor()]))
test = datasets.MNIST("", train=False, download=True,
transform = transforms.Compose([transforms.ToTensor()]))
trainset = torch.utils.data.DataLoader(train, batch_size=15, shuffle=True)
testset = torch.utils.data.DataLoader(test, batch_size=15, shuffle=True)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment