burn::data::dataloader

Module split

Expand description

Module to split a dataloader.

Functions§

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