tfep.io.sampler

A PyTorch stateful batch sampler that can be used to correctly resume training mid-epoch.

See documentation of the class StatefulBatchSampler for more details.

Classes

StatefulBatchSampler(dataset[, batch_size, ...])

A PyTorch stateful batch sampler to resume training mid-epoch.