Skip to content

Instantly share code, notes, and snippets.

@miki998

miki998/batch.py

Created April 13, 2020 10:23
Show Gist options
  • Save miki998/a264f63d7558d30b8d16936c4a61e2b6 to your computer and use it in GitHub Desktop.
Save miki998/a264f63d7558d30b8d16936c4a61e2b6 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
batch.py
# batch_size, epoch and iteration
batch_size = 100
# Pytorch train and test sets
train = torch.utils.data.TensorDataset(train_x,y_train)
test = torch.utils.data.TensorDataset(X_cv,y_cv)
# data loader
train_loader = torch.utils.data.DataLoader(train, batch_size = batch_size, shuffle = False)
test_loader = torch.utils.data.DataLoader(test, batch_size = batch_size, shuffle = False)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment