draugr.torch_utilities.generators.batching

Classes

LimitedBatchResampler(batch_sampler, ...[, ...])

Wraps a BatchSampler, re-sampling from it until a specified number of iterations have been sampled