Update conformer.py

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

View File

@ -33,7 +33,7 @@ class Conformer(Transformer):
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 d_model (int): attention 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
num_decoder_layers (int): number of decoder layers num_decoder_layers (int): number of decoder layers
dropout (float): dropout rate dropout (float): dropout rate