multiml.task.pytorch.samplers package

Submodules

• multiml.task.pytorch.samplers.simple_batch_sampler module
  • SimpleBatchSampler
    • SimpleBatchSampler.__init__()
    • SimpleBatchSampler.__len__()

Module contents

class multiml.task.pytorch.samplers.SimpleBatchSampler(num_samples, batch_size, shuffle, device='cpu')

Bases: Sampler

__init__(num_samples, batch_size, shuffle, device='cpu')

__len__()