fix no_pos call

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}")
 

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: