Method default_config

axlearn/audio/decoder_asr.py:1144–1148

(cls)

    @classmethod
    def default_config(cls) -> Config:
        cfg = super().default_config()
        transformer_cfg = cfg.decoder.transformer.layer
        transformer_cfg.cross_attention = TransformerAttentionLayer.default_config()
        return cfg

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