Add wenet ref in ConvolutionModule class.

This commit is contained in:
yaozengwei 2022-04-11 12:38:45 +08:00
parent 1d74c5e596
commit 6c1f9b5181

View File

@ -1410,6 +1410,7 @@ class ConvolutionModule(nn.Module):
bias=bias,
)
# from https://github.com/wenet-e2e/wenet/blob/main/wenet/transformer/convolution.py # noqa
if causal:
self.left_padding = kernel_size - 1
padding = 0