DataloaderHparams

class composer.datasets.DataloaderHparams(num_workers: int, prefetch_factor: int, persistent_workers: bool, pin_memory: bool, timeout: int)

Bases: yahp.hparams.Hparams

Hyperparameters to initialize a torch.utils.data.Dataloader.

initialize_object(batch_size: int, sampler: torch.utils.data.sampler.Sampler, dataloader_spec: DataloaderSpec) → DataLoader

Initializes the dataloader.