s2t: fix encoder.py

2 years ago · cdcb1a5316
parent ed2819d7af
commit cdcb1a5316
1 changed files with 1 additions and 0 deletions
--- a/paddlespeech/s2t/modules/encoder.py
+++ b/paddlespeech/s2t/modules/encoder.py
@ -255,6 +255,7 @@ class BaseEncoder(nn.Layer):
                xs,
                att_mask,
                pos_emb,
+                mask_pad=paddle.ones([0, 0, 0], dtype=paddle.bool),
                att_cache=att_cache[i:i + 1] if elayers > 0 else att_cache,
                cnn_cache=cnn_cache[i:i + 1]
                if paddle.shape(cnn_cache)[0] > 0 else cnn_cache, )