multiml.task.pytorch.samplers.simple_batch_sampler module

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

Bases: Sampler