paddlespeech.s2t.io.sampler module

class paddlespeech.s2t.io.sampler.SortagradBatchSampler(dataset, batch_size, shuffle=False, drop_last=False, sortagrad=False, shuffle_method='batch_shuffle')

Bases: BatchSampler

class paddlespeech.s2t.io.sampler.SortagradDistributedBatchSampler(dataset, batch_size, num_replicas=None, rank=None, shuffle=False, drop_last=False, sortagrad=False, shuffle_method='batch_shuffle')

Bases: DistributedBatchSampler

Methods

set_epoch(epoch)

Sets the epoch number.