Skip to content

Instantly share code, notes, and snippets.

@pranshuj73

pranshuj73/script.py

Created July 3, 2020 11:10
Show Gist options
  • Save pranshuj73/f6ada63e48ce04cb2491d06750010f72 to your computer and use it in GitHub Desktop.
Save pranshuj73/f6ada63e48ce04cb2491d06750010f72 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
script.py
batch_num = 400
train_dl = DataLoader(train_data, batch_num, shuffle=True, num_workers=4, pin_memory=True)
val_dl = DataLoader(val_data, batch_num*2, num_workers=4, pin_memory=True)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment