Skip to content

Instantly share code, notes, and snippets.

@omarsar
Created October 15, 2018 09:09
Show Gist options
  • Save omarsar/86e18d457293f31bb2513ffae90b0c99 to your computer and use it in GitHub Desktop.
Save omarsar/86e18d457293f31bb2513ffae90b0c99 to your computer and use it in GitHub Desktop.
train_loader = DataLoader(gmdataset_train, batch_size=16,
shuffle=True, pin_memory=True,
collate_fn=mt_datasets.mt_collate,
num_workers=1)
val_loader = DataLoader(gmdataset_val, batch_size=16,
shuffle=True, pin_memory=True,
collate_fn=mt_datasets.mt_collate,
num_workers=1)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment