Skip to content

Instantly share code, notes, and snippets.

@kabirahuja2431

kabirahuja2431/dataloader.py

Last active October 4, 2019 10:41
Show Gist options
  • Save kabirahuja2431/aec470875de24868b538b94124640de0 to your computer and use it in GitHub Desktop.
Save kabirahuja2431/aec470875de24868b538b94124640de0 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
dataloader.py
from torch.utils.data import DataLoader
#Create a dataset object
dataset = CutsomDataset('path_to/somefile')
#Wrap it around a dataloader
dataloader = DataLoader(dataset, batch_size = 64, num_workers = 5)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment