Update conformer.py

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

View File

@ -35,7 +35,7 @@ class Conformer(Transformer):
subsampling_factor (int): subsampling factor of encoder (the convolution layers before transformers)
d_model (int): attention dimension
nhead (int): number of head
dim_feedforward (int): feedforward dimention
dim_feedforward (int): feedforward dimension
num_encoder_layers (int): number of encoder layers
dropout (float): dropout rate
cnn_module_kernel (int): Kernel size of convolution module