fix autograd maybe has problem when using inplace operation

5 years ago · 12c01f39aa
parent 220c9443a9
commit 12c01f39aa
1 changed files with 2 additions and 2 deletions
--- a/deepspeech/modules/conformer_convolution.py
+++ b/deepspeech/modules/conformer_convolution.py
@ -122,7 +122,7 @@ class ConvolutionModule(nn.Layer):

        # mask batch padding
        if mask_pad is not None:
-            x.masked_fill_(mask_pad, 0.0)
+            x = x.masked_fill(mask_pad, 0.0)

        if self.lorder > 0:
            if cache is None:
@ -156,7 +156,7 @@ class ConvolutionModule(nn.Layer):

        # mask batch padding
        if mask_pad is not None:
-            x.masked_fill_(mask_pad, 0.0)
+            x = x.masked_fill(mask_pad, 0.0)

        x = x.transpose([0, 2, 1])  # [B, T, C]
        return x, new_cache