pytorch_pfn_extras.dataloaders.dataloader

Functions

pytorch_pfn_extras.dataloaders.dataloader.default_collate(batch)	Take in a batch of data and put the elements within the batch into a tensor with an additional outer dimension - batch size.
pytorch_pfn_extras.dataloaders.dataloader.default_convert(data)	Convert each NumPy array element into a torch.Tensor.
pytorch_pfn_extras.dataloaders.dataloader.get_worker_info()	Returns the information about the current DataLoader iterator worker process.

Classes

pytorch_pfn_extras.dataloaders.dataloader.DataLoader(dataset)

Data loader combines a dataset and a sampler, and provides an iterable over the given dataset.