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 |