don't use +=

2020-08-14 00:48:27 +02:00 · 2020-08-14 00:48:27 +02:00 · 77c577ddab
parent 383c5f7185
commit 77c577ddab
1 changed files with 1 additions and 1 deletions
--- a/mozilla_voice_tts/tts/layers/glow_tts/encoder.py
+++ b/mozilla_voice_tts/tts/layers/glow_tts/encoder.py
@ -108,7 +108,7 @@ class Encoder(nn.Module):
        # pass embedding layer
        # [B ,T, D]
        x = self.emb(x) * math.sqrt(self.hidden_channels)
-        x += self.pe[:x.shape[1]].unsqueeze(0)
+        x = x + self.pe[:x.shape[1]].unsqueeze(0)
        # [B, D, T]
        x = torch.transpose(x, 1, -1)
        # compute input sequence mask