Skip to content

Instantly share code, notes, and snippets.

@sadimanna
Created June 30, 2021 10:29
Show Gist options
  • Select an option

  • Save sadimanna/8f3a9094f8284dc5340d45fa3ef812a5 to your computer and use it in GitHub Desktop.

Select an option

Save sadimanna/8f3a9094f8284dc5340d45fa3ef812a5 to your computer and use it in GitHub Desktop.
dg = C10DataGen('train',trimages)
dl = DataLoader(dg,batch_size = 128,drop_last=True)
vdg = C10DataGen('valid',valimages)
vdl = DataLoader(vdg,batch_size = 128,drop_last=True)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment