espnet2.samplers.build_batch_sampler.build_category_batch_sampler

source

espnet2.samplers.build_batch_sampler.build_category_batch_sampler(type: str, batch_size: int | None = None, batch_bins: int | None = None, shape_files: Tuple[str, ...] | List[str] | None = None, min_batch_size: int | None = None, max_batch_size: int | None = None, upsampling_factor: float | None = None, category_upsampling_factor: float | None = None, dataset_upsampling_factor: float | None = None, dataset_scaling_factor: float | None = None, drop_last: bool = False, category2utt_file: str | None = None, dataset2utt_parent_dir: str | None = None, epoch: int = 1, num_batches: int | None = None, distributed: bool = False) → Tuple[AbsSampler, dict]