flair.datasets.base.DataLoader

class flair.datasets.base.DataLoader(dataset, batch_size=1, shuffle=False, sampler=None, batch_sampler=None, drop_last=False, timeout=0, worker_init_fn=None)View on GitHub

Bases: DataLoader

__init__(dataset, batch_size=1, shuffle=False, sampler=None, batch_sampler=None, drop_last=False, timeout=0, worker_init_fn=None)View on GitHub

Methods

__init__(dataset[, batch_size, shuffle, ...])
check_worker_number_rationality()

Attributes

multiprocessing_context
dataset
batch_size
num_workers
pin_memory
drop_last
timeout
sampler
pin_memory_device
prefetch_factor