Dataloader¤
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 |