burn::data::dataloader::split

Function split_dataloader

pub fn split_dataloader<B, O>(
    dataloader: Arc<dyn DataLoader<B, O>>,
    devices: &[<B as Backend>::Device],
) -> Vec<Arc<dyn DataLoader<B, O>>>
where
    B: Backend,
Expand description

Splits a dataloader into multiple partial dataloaders (one per device).