Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

convolution encoder with GLU and res connections

This commit is contained in:
erogol 2020-08-13 12:42:48 +02:00
parent 498a3ea36f
commit 45fbc0d003
2 changed files with 10 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
TTS/tts/layers/glow_tts/glow.py
View File

@ -65,7 +65,7 @@ class ConvLayerNorm(nn.Module):
self.proj.bias.data.zero_() self.proj.bias.data.zero_()
def forward(self, x, x_mask): def forward(self, x, x_mask):
x_org = x x_res = x
for i in range(self.num_layers): for i in range(self.num_layers):
x = self.conv_layers[i](x * x_mask) x = self.conv_layers[i](x * x_mask)
x = self.norm_layers[i](x * x_mask) x = self.norm_layers[i](x * x_mask)

23
TTS/tts/models/glow_tts.py
View File

@ -69,20 +69,15 @@ class GlowTts(nn.Module):
self.length_scale=1. self.length_scale=1.
self.encoder = Encoder(num_chars, self.encoder = Encoder(num_chars,
out_channels, out_channels=out_channels,
hidden_channels_enc or hidden_channels, hidden_channels=hidden_channels,
filter_channels, filter_channels=filter_channels,
filter_channels_dp, filter_channels_dp=filter_channels_dp,
encoder_type, num_layers=num_layers_enc,
num_heads, kernel_size=kernel_size,
num_layers_enc, dropout_p=dropout_p,
kernel_size, mean_only=mean_only,
dropout_p, c_in_channels=c_in_channels)
rel_attn_window_size=rel_attn_window_size,
input_length=input_length,
mean_only=mean_only,
use_prenet=True,
c_in_channels=c_in_channels)
self.decoder = Decoder(out_channels, self.decoder = Decoder(out_channels,
hidden_channels_dec or hidden_channels, hidden_channels_dec or hidden_channels,