From cda38ae4df75c0be717278d93ff0cd60013afd28 Mon Sep 17 00:00:00 2001 From: Vahid Date: Thu, 22 Oct 2020 00:47:23 -0700 Subject: [PATCH] fixed style. Signed-off-by: Vahid --- nemo/collections/asr/modules/conformer_modules.py | 1 + 1 file changed, 1 insertion(+) diff --git a/nemo/collections/asr/modules/conformer_modules.py b/nemo/collections/asr/modules/conformer_modules.py index 8e6e0a0fefbc..5662ddbfe439 100644 --- a/nemo/collections/asr/modules/conformer_modules.py +++ b/nemo/collections/asr/modules/conformer_modules.py @@ -70,6 +70,7 @@ class ConformerFeedForward(nn.Module): """ feed-forward module of Conformer model. """ + def __init__(self, d_model, d_ff, dropout, activation=Swish()): super(ConformerFeedForward, self).__init__() self.linear1 = nn.Linear(d_model, d_ff)