espnet2.asr_transducer.normalization.get_normalization

source

espnet2.asr_transducer.normalization.get_normalization(normalization_type: str, eps: float | None = None, partial: float | None = None) → Tuple[Module, Dict]

Normalization modules for Transducer.

This module provides various normalization techniques used in transducer models. It includes basic normalization, layer normalization, RMS normalization, and scale normalization. The get_normalization function retrieves the appropriate normalization module based on the specified type.

Functions: : get_normalization: Get normalization module and arguments given parameters.

Classes: : BasicNorm: Basic normalization module. RMSNorm: RMS normalization module. ScaleNorm: Scale normalization module.