Skip to content

Instantly share code, notes, and snippets.

@williamFalcon

williamFalcon/9t_2.py

Created July 29, 2019 17:45
Show Gist options
  • Save williamFalcon/f2e989c981205308defd70a91c9a7cdd to your computer and use it in GitHub Desktop.
Save williamFalcon/f2e989c981205308defd70a91c9a7cdd to your computer and use it in GitHub Desktop.
Download ZIP
Raw
9t_2.py
dataset = MNIST(root=self.hparams.data_root, train=train, download=True)
loader = DataLoader(dataset, batch_size=32, shuffle=True)
for batch in loader:
x, y = batch
model.training_step(x, y)
...
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment