tensorplay.utils.data.dataloader
Classes
class DataLoader [source]
python
DataLoader(dataset, batch_size=1, shuffle=False, sampler=None, batch_sampler=None, num_workers=0, collate_fn=None, pin_memory=False, drop_last=False, timeout=0, worker_init_fn=None, multiprocessing_context=None, generator=None, *, prefetch_factor=2, persistent_workers=False, device=None, worker_debug=False)Methods
__init__(self, dataset, batch_size=1, shuffle=False, sampler=None, batch_sampler=None, num_workers=0, collate_fn=None, pin_memory=False, drop_last=False, timeout=0, worker_init_fn=None, multiprocessing_context=None, generator=None, *, prefetch_factor=2, persistent_workers=False, device=None, worker_debug=False) [source]
Initialize self. See help(type(self)) for accurate signature.
class WorkerInfo [source]
python
WorkerInfo(**kwargs)Methods
__init__(self, **kwargs) [source]
Initialize self. See help(type(self)) for accurate signature.
Functions
default_collate() [source]
python
default_collate(batch)Puts each data field into a tensor with outer dimension batch size
get_worker_info() [source]
python
get_worker_info()