From 61235c3b832ee5ac7ef571f10b63c87f5dc64dce Mon Sep 17 00:00:00 2001
From: Hui Zhang <zhtclz@foxmail.com>
Date: Sun, 24 Oct 2021 13:58:45 +0000
Subject: [PATCH] fix no_pos call

---
 deepspeech/models/lm/transformer.py | 3 +--
 deepspeech/modules/encoder.py       | 2 +-
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/deepspeech/models/lm/transformer.py b/deepspeech/models/lm/transformer.py
index 9392a15ed..dcae4ea0f 100644
--- a/deepspeech/models/lm/transformer.py
+++ b/deepspeech/models/lm/transformer.py
@@ -45,8 +45,7 @@ class TransformerLM(nn.Layer, LMInterface, BatchScorerInterface):
         if pos_enc == "sinusoidal":
             pos_enc_layer_type = "abs_pos"
         elif pos_enc is None:
-            #TODO
-            pos_enc_layer_type = "None"
+            pos_enc_layer_type = "no_pos"
         else:
             raise ValueError(f"unknown pos-enc option: {pos_enc}")
 
diff --git a/deepspeech/modules/encoder.py b/deepspeech/modules/encoder.py
index bbe822af6..a9b5e8a6a 100644
--- a/deepspeech/modules/encoder.py
+++ b/deepspeech/modules/encoder.py
@@ -384,7 +384,7 @@ class TransformerEncoder(BaseEncoder):
 
         Args:
             xs (paddle.Tensor): Input tensor. (B, T, D)
-            masks (paddle.Tensor): Mask tensor. (B, 1, T)
+            masks (paddle.Tensor): Mask tensor. (B, T, T)
             cache (List[paddle.Tensor]): List of cache tensors.
 
         Returns: