Update conformer.py

feedforward dimention -> feedforward dimension
This commit is contained in:
Seonuk Kim 2025-01-06 11:16:44 +09:00 committed by GitHub
parent d0c4081ab5
commit 87d5ee4726
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -42,7 +42,7 @@ class Conformer(EncoderInterface):
subsampling_factor (int): subsampling factor of encoder (the convolution layers before transformers) subsampling_factor (int): subsampling factor of encoder (the convolution layers before transformers)
d_model (int): attention dimension, also the output dimension d_model (int): attention dimension, also the output dimension
nhead (int): number of head nhead (int): number of head
dim_feedforward (int): feedforward dimention dim_feedforward (int): feedforward dimension
num_encoder_layers (int): number of encoder layers num_encoder_layers (int): number of encoder layers
dropout (float): dropout rate dropout (float): dropout rate
layer_dropout (float): layer-dropout rate. layer_dropout (float): layer-dropout rate.