espnet2.asr_transducer.encoder.building.build_ebranchformer_block

source

espnet2.asr_transducer.encoder.building.build_ebranchformer_block(configuration: List[Dict[str, Any]], main_params: Dict[str, Any]) → EBranchformer

Build E-Branchformer block.

This function constructs an E-Branchformer block, which is part of the encoder architecture for transducers. It utilizes various configurations and main parameters to create a functional block for encoding input data.

• Parameters:
  • configuration – A list of dictionaries containing the E-Branchformer block configuration parameters such as hidden_size, linear_size, dropout_rate, pos_wise_dropout_rate, conv_mod_kernel_size, heads, att_dropout_rate, depth_conv_kernel_size, norm_eps, and norm_partial.
  • main_params – A dictionary containing the main encoder parameters, including activation functions, normalization types, and dropout rates.
• Returns: A callable that constructs an E-Branchformer block : function when invoked.
• Return type:EBranchformer

Examples

config = [{ : “hidden_size”: 256, “linear_size”: 512, “dropout_rate”: 0.1, “pos_wise_dropout_rate”: 0.1, “conv_mod_kernel_size”: 3, “heads”: 4, “att_dropout_rate”: 0.1, “depth_conv_kernel_size”: 3, “norm_eps”: 1e-5, “norm_partial”: None

}] main_params = build_main_parameters() e_branchformer_block = build_ebranchformer_block(config, main_params)()