Class Config

axlearn/audio/decoder_asr.py:287–293

Configures CTCDecoderModel.

    @config_class
    class Config(BaseASRDecoderModel.Config):
        """Configures CTCDecoderModel."""

        # Layer to map hidden state to vocab logits.
        lm_head: BaseLayer.Config = Linear.default_config()
        # Blank token ID.
        blank_id: int = 0

    def __init__(self, cfg: Config, *, parent: Optional[Module]):
        super().__init__(cfg, parent=parent)