espnet2.asr.partially_AR_model.PartiallyARInference

source

class espnet2.asr.partially_AR_model.PartiallyARInference(ctc: CTC, decoder: AbsDecoder, threshold_probability: float, sos: int | None = None, eos: int | None = None, mask_token: int | None = None, token_list: List[int] | None = None, scorers: Dict[str, ScorerInterface] | None = None, weights: Dict[str, float] | None = None, beam_size: int = 10, max_seq_len: int = 5, max_mask_parallel: int = -1)

Bases: Module

Mask-CTC-based partially autoregressive inference

Initialize Mask-CTC inference

forward(enc_out: Tensor, *args, **kwargs) → List[Hypothesis]

Perform Semi-AR inference

set_hyp_primer(primer: List[int])