From 77c577ddab6cc803c9dbb4a58be003ffab4dfaf3 Mon Sep 17 00:00:00 2001 From: erogol Date: Fri, 14 Aug 2020 00:48:27 +0200 Subject: [PATCH] don't use += --- mozilla_voice_tts/tts/layers/glow_tts/encoder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mozilla_voice_tts/tts/layers/glow_tts/encoder.py b/mozilla_voice_tts/tts/layers/glow_tts/encoder.py index 1a1fe2ca..d8bbc07c 100644 --- 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