Merge pull request #791 from Jackwaterveg/fix_bug

Fix bug of paddling len for "def forward_chunk_by_chunk"
3 years ago · b874d84671
parent 794294e9cb 2b3b985227
commit b874d84671
1 changed files with 4 additions and 2 deletions
--- a/deepspeech/models/ds2_online/deepspeech2.py
+++ b/deepspeech/models/ds2_online/deepspeech2.py
@ -180,8 +180,10 @@ class CRNNEncoder(nn.Layer):

        eouts_chunk_list = []
        eouts_chunk_lens_list = []
-
-        padding_len = chunk_stride - (max_len - chunk_size) % chunk_stride
+        if (max_len - chunk_size) % chunk_stride != 0:
+            padding_len = chunk_stride - (max_len - chunk_size) % chunk_stride
+        else:
+            padding_len = 0
        padding = paddle.zeros((x.shape[0], padding_len, x.shape[2]))
        padded_x = paddle.concat([x, padding], axis=1)
        num_chunk = (max_len + padding_len - chunk_size) / chunk_stride + 1