Skip to content

Dataloader¤


source

BaseDataLoader¤

BaseDataLoader (dataset, batch_size:int=1, shuffle:bool=False, num_workers:int=0, drop_last:bool=False, generator:Union[ jax_dataloader.utils.Generator,jax.Array,ForwardRef('torc h.Generator'),NoneType]=None, **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
generator Optional[GeneratorType] None
kwargs