Skip to content

Instantly share code, notes, and snippets.

@jguertl

jguertl/DataLoader

Created March 13, 2019 15:38
Show Gist options
  • Star 1 You must be signed in to star a gist
  • Fork 1 You must be signed in to fork a gist
  • Save jguertl/ac7d663316cda6e46d208893d67f187d to your computer and use it in GitHub Desktop.
Save jguertl/ac7d663316cda6e46d208893d67f187d to your computer and use it in GitHub Desktop.
Download ZIP
Raw
DataLoader
data_loader = mx.gluon.data.DataLoader(dataset, batch_size=5)
for X_batch, y_batch in data_loader:
print("X_batch has shape {}, and y_batch has shape {}".format(X_batch.shape, y_batch.shape))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment