mirror of https://github.com/coqui-ai/TTS.git
bug fix
This commit is contained in:
parent
b11cf66c69
commit
b73d501c7d
|
@ -285,12 +285,7 @@ class Decoder(nn.Module):
|
||||||
if greedy:
|
if greedy:
|
||||||
memory_input = outputs[-1]
|
memory_input = outputs[-1]
|
||||||
else:
|
else:
|
||||||
combine prev. model output and prev. real target
|
memory_input = memory[t-1]
|
||||||
memory_input = torch.div(outputs[-1] + memory[t-1], 2.0)
|
|
||||||
noise = torch.autograd.Variable(
|
|
||||||
memory_input.data.new(memory_input.size()).normal_(0.0, 2.0))
|
|
||||||
memory_input = memory_input + noise
|
|
||||||
|
|
||||||
# Prenet
|
# Prenet
|
||||||
processed_memory = self.prenet(memory_input)
|
processed_memory = self.prenet(memory_input)
|
||||||
# Attention RNN
|
# Attention RNN
|
||||||
|
|
Loading…
Reference in New Issue