espnet2.samplers.sorted_batch_sampler.SortedBatchSampler

source

class espnet2.samplers.sorted_batch_sampler.SortedBatchSampler(batch_size: int, shape_file: str, sort_in_batch: str = 'descending', sort_batch: str = 'ascending', drop_last: bool = False)

Bases: AbsSampler

BatchSampler with sorted samples by length.

• Parameters:
  • batch_size
  • shape_file
  • sort_in_batch – ‘descending’, ‘ascending’ or None.
  • sort_batch