cerebras.modelzoo.data.common.h5_map_dataset.samplers#
Classes
Handle shuffling and skipping |
|
A slight modification of the PyTorch batch sampler such that any samples not yielded at the end of an epoch when drop_last=True will be yielded at the start of the next epoch. |
|
A sampler to handle sharding, batching, and skipping of map style datasets intended for use on CSX. |
|