Update conformer.py

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

View File

@ -32,7 +32,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
num_decoder_layers (int): number of decoder layers
dropout (float): dropout rate