fix conformer multi-gpu training

3 years ago · 6ac47f4c57
parent bb82745f19
commit 6ac47f4c57
1 changed files with 1 additions and 1 deletions
--- a/paddlespeech/s2t/modules/encoder.py
+++ b/paddlespeech/s2t/modules/encoder.py
@ -329,7 +329,7 @@ class BaseEncoder(nn.Layer):
            chunk_xs = xs[:, cur:end, :]

            (y, att_cache, cnn_cache) = self.forward_chunk(
-                chunk_xs, offset, required_cache_size, att_cache, cnn_cache)
+                chunk_xs, offset, required_cache_size, att_cache, cnn_cache, paddle.ones([0,0,0], dtype=paddle.bool))

            outputs.append(y)
            offset += y.shape[1]