Skip to content

LavenderDataLoader - Collater

Collaters control how individual samples are combined into batches.

dataloader = LavenderDataLoader(
dataset_id=dataset.id,
shardsets=[shardset.id],
collater="pylist", # To use a module without parameters, specify the module name only
batch_size=10,
)