paddlespeech.s2t.io.sampler 模块

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

基础: 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')

基本： DistributedBatchSampler

方法

set_epoch(epoch)

设置纪元编号。