espnet2.asr_transducer.utils.get_convinput_module_parameters

source

espnet2.asr_transducer.utils.get_convinput_module_parameters(input_size: int, last_conv_size, subsampling_factor: int, is_vgg: bool = True) → Tuple[Tuple[int, int] | int, int]

Return the convolution module parameters.

• Parameters:
  • input_size – Module input size.
  • last_conv_size – Last convolution size for module output size computation.
  • subsampling_factor – Total subsampling factor.
  • is_vgg – Whether the module type is VGG-like.
• Returns: First MaxPool2D kernel size or second Conv2d kernel size and stride. output_size: Convolution module output size.