don't use +=

2026-09-11 12:00:24 +08:00 · 2020-09-11 19:01:38 +02:00
parent 383c5f7185
commit 77c577ddab
1 changed files with 1 additions and 1 deletions
@@ -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