class
torch.utils.data.
DataLoader
(dataset,
batch_size=1,
shuffle=False,
sampler=None,
batch_sampler=None,
num_workers=0,
collate_fn=<function default_collate>,
pin_memory=False,
drop_last=False)
[source]
Data loader. Combines a dataset and a sampler, and providessingle- or multi-process iterators over the dataset.
Parameters: |
|
---|