Skip to content

Instantly share code, notes, and snippets.

@omarsar

omarsar/dimensions.py

Created August 31, 2019 13:54
Show Gist options
  • Save omarsar/902971688e72c06687bdd2d188605a68 to your computer and use it in GitHub Desktop.
Save omarsar/902971688e72c06687bdd2d188605a68 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
dimensions.py
for images, labels in trainloader:
print("Image batch dimensions:", images.shape)
print("Image label dimensions:", labels.shape)
break
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment