burn_core::data::dataloader

Module split

Available on crate features std and dataset only.

Expand description

Module to split a dataloader.

Functions§

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