pytorch_pfn_extras.dataloaders.dataloader#

Functions

pytorch_pfn_extras.dataloaders.dataloader.default_collate(batch)

Function that takes in a batch of data and puts the elements within the batch into a tensor with an additional outer dimension - batch size.

pytorch_pfn_extras.dataloaders.dataloader.default_convert(data)

Function that converts 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.