espnet2.asr.maskctc_model.MaskCTCInference

source

class espnet2.asr.maskctc_model.MaskCTCInference(asr_model: MaskCTCModel, n_iterations: int, threshold_probability: float)

Bases: Module

Mask-CTC-based non-autoregressive inference

Initialize Mask-CTC inference

forward(enc_out: Tensor) → List[Hypothesis]

Perform Mask-CTC inference

ids2text(ids: List[int])