espnet2.train.collate_fn.HuBERTCollateFn

source

class espnet2.train.collate_fn.HuBERTCollateFn(float_pad_value: float | int = 0.0, int_pad_value: int = -32768, label_downsampling: int = 1, pad: bool = False, rand_crop: bool = True, crop_audio: bool = True, not_sequence: Collection[str] = (), window_size: float = 25, window_shift: float = 20, sample_rate: float = 16, noise_scp: str = 'data/noise/wav.scp', noise_apply_prob: float = 1.0, noise_db_range: str = '-5_20', dynamic_mixing_gain_db: float = 5.0, dynamic_mixing_prob=0.1, mix_speech: bool = False, reverb_speech: bool = False, rir_scp: str = 'data/rirs/wav.scp', rir_apply_prob: float = 0.3, train: bool = True)

Bases: CommonCollateFn

Functor class of common_collate_fn()