sketchgraphs_models.nn.data_util

Utilities and extensions to work with the torch.utils.data package.

Classes

MultiEpochSampler(batch_sampler, num_epochs)

A sampler wrapper which creates multiple epochs of indices from one sampler.