Closed jsalotti closed 1 year ago
General description of your pull request with the list of new features and/or bugs. Closes #107
New feature 1 : Sampler for train loader can be constructed before calling the method
sampler = torch.utils.data.RandomSampler(dataset, replacement=True) loader = dataset.train_loader(sampler=sampler)
New feature 2 : Sampler kwargs can be given to train_loader method:
train_loader
sampler = torch.utils.data.RandomSampler loader = dataset.train_loader(sampler=sampler, sampler_kwargs={"replacement":True})
This pull request includes
[ ] Bug fix (non-breaking change which fixes an issue)
[x] New feature (non-breaking change which adds functionality)
[ ] Breaking change (fix or feature that would cause existing functionality to not work as expected)
[ ] This change requires a documentation update
General description of your pull request with the list of new features and/or bugs. Closes #107
New feature 1 : Sampler for train loader can be constructed before calling the method
New feature 2 : Sampler kwargs can be given to
train_loader
method:This pull request includes
[ ] Bug fix (non-breaking change which fixes an issue)
[x] New feature (non-breaking change which adds functionality)
[ ] Breaking change (fix or feature that would cause existing functionality to not work as expected)
[ ] This change requires a documentation update