Skip to content

Instantly share code, notes, and snippets.

@Noob-can-Compile

Noob-can-Compile/Batch_data.py

Created December 1, 2019 12:06
Show Gist options
  • Save Noob-can-Compile/8a35f7e16e7b44368b5351aa3d59b8d5 to your computer and use it in GitHub Desktop.
Save Noob-can-Compile/8a35f7e16e7b44368b5351aa3d59b8d5 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Batch_data.py
# load training data in batches
batch_size = 10
train_loader = DataLoader(transformed_dataset,
batch_size=batch_size,
shuffle=True,
num_workers=0)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment