提交 2b3b9852 编写于 作者: H huangyuxin

fix paddling len

上级 03a50d7b
...@@ -180,8 +180,10 @@ class CRNNEncoder(nn.Layer): ...@@ -180,8 +180,10 @@ class CRNNEncoder(nn.Layer):
eouts_chunk_list = [] eouts_chunk_list = []
eouts_chunk_lens_list = [] eouts_chunk_lens_list = []
if (max_len - chunk_size) % chunk_stride != 0:
padding_len = chunk_stride - (max_len - chunk_size) % chunk_stride 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])) padding = paddle.zeros((x.shape[0], padding_len, x.shape[2]))
padded_x = paddle.concat([x, padding], axis=1) padded_x = paddle.concat([x, padding], axis=1)
num_chunk = (max_len + padding_len - chunk_size) / chunk_stride + 1 num_chunk = (max_len + padding_len - chunk_size) / chunk_stride + 1
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册