mirror of
https://github.com/wassname/TTS.git
synced 2026-10-27 12:30:27 +08:00
don't use +=
This commit is contained in:
1 parent
383c5f7185
commit
77c577ddab
1 file changed
+1
-1
@@ -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
|
||||
|
||||
Reference in new issue
Block a user