added pos_emb_max_len.

Signed-off-by: Vahid <vnoroozi@nvidia.com>
NVIDIA · Nov 2, 2020 · 306ddf6 · 306ddf6
1 parent 58f9588
commit 306ddf6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/nemo/collections/asr/modules/conformer_encoder.py b/nemo/collections/asr/modules/conformer_encoder.py
@@ -153,7 +153,11 @@ def __init__(
 
         if self_attention_model == "rel_pos":
             self.pos_enc = RelPositionalEncoding(
-                d_model=d_model, dropout_rate=dropout, max_len=pos_emb_max_len, xscale=self.xscale, dropout_emb_rate=dropout_emb
+                d_model=d_model,
+                dropout_rate=dropout,
+                max_len=pos_emb_max_len,
+                xscale=self.xscale,
+                dropout_emb_rate=dropout_emb,
             )
         elif self_attention_model == "abs_pos":
             self.pos_enc = PositionalEncoding(