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]

Get normalization module and arguments given parameters.

• Parameters:
  • normalization_type – Normalization module type.
  • eps – Value added to the denominator.
  • partial – Value defining the part of the input used for RMS stats (RMSNorm).
• Returns: Normalization module class : Normalization module arguments