fix multi-gpu training

3 years ago · 285d76176a
parent 6ac47f4c57
commit 285d76176a
1 changed files with 2 additions and 1 deletions
--- a/paddlespeech/s2t/modules/encoder.py
+++ b/paddlespeech/s2t/modules/encoder.py
@ -329,7 +329,8 @@ 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, paddle.ones([0,0,0], dtype=paddle.bool))
+                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]