parsnip.ParsnipModel.get_data_loader
- ParsnipModel.get_data_loader(dataset, augment=False, **kwargs)
Get a PyTorch DataLoader for an lcdata Dataset
- Parameters:
dataset (
Dataset) – Dataset to loadaugment (bool, optional) – Whether to augment the dataset, by default False
- Returns:
PyTorch DataLoader for the dataset
- Return type:
DataLoader