Skip to content

Dataloader¤


source

BaseDataLoader¤

 BaseDataLoader (dataset, batch_size:int=1, shuffle:bool=False,
                 num_workers:int=0, drop_last:bool=False, **kwargs)

Dataloader Interface

Type Default Details
dataset
batch_size int 1 batch size
shuffle bool False if true, dataloader shuffles before sampling each batch
num_workers int 0 how many subprocesses to use for data loading.
drop_last bool False
kwargs