Skip to content

Instantly share code, notes, and snippets.

@omarsar

omarsar/one_batch.py

Last active August 24, 2019 17:57
Show Gist options
  • Save omarsar/4f660c2b2f3e4f284c112fc6324989a5 to your computer and use it in GitHub Desktop.
Save omarsar/4f660c2b2f3e4f284c112fc6324989a5 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
one_batch.py
## test the model with 1 batch
model = MyModel()
for images, labels in trainloader:
print("batch size:", images.shape)
out = model(images)
print(out.shape)
break
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment