espnet2.samplers.category_balanced_sampler.CategoryBalancedSampler

source

class espnet2.samplers.category_balanced_sampler.CategoryBalancedSampler(batch_size: int, min_batch_size: int = 1, drop_last: bool = False, category2utt_file: str | None = None, epoch: int = 1, **kwargs)

Bases: AbsSampler