Module split

Expand description

Module to split a dataloader.

Functionsยง

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