Skip to content

Instantly share code, notes, and snippets.

@albertlai431

albertlai431/dataset.py

Last active December 4, 2018 01:57
Show Gist options
  • Save albertlai431/b851266f1efd0b0898900d507199ab68 to your computer and use it in GitHub Desktop.
Save albertlai431/b851266f1efd0b0898900d507199ab68 to your computer and use it in GitHub Desktop.
Download ZIP
Loading Dataset
Raw
dataset.py
#Loading dataset
transform = transforms.Compose([transforms.ToTensor(),
transforms.Normalize((0.5,0.5,0.5), (0.5,0.5,0.5))])
train_dataset = datasets.FashionMNIST(root='./data',
train=True,
download=True,
transform=transform)
test_dataset = datasets.FashionMNIST(root='./data',
train=False,
download=True,
transform=transform)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment