mirror of https://github.com/coqui-ai/TTS.git
Merge branch 'master' of github.com:mozilla/TTS
This commit is contained in:
commit
3a72d75ecd
|
@ -371,7 +371,7 @@ class Decoder(nn.Module):
|
||||||
greedy = not self.training
|
greedy = not self.training
|
||||||
if memory is not None:
|
if memory is not None:
|
||||||
memory = self._reshape_memory(memory)
|
memory = self._reshape_memory(memory)
|
||||||
T_decoder = memory.size(0)
|
T_decoder = memory.size(0)
|
||||||
# go frame as zeros matrix
|
# go frame as zeros matrix
|
||||||
initial_memory = inputs.data.new(B, self.memory_dim * self.r).zero_()
|
initial_memory = inputs.data.new(B, self.memory_dim * self.r).zero_()
|
||||||
# decoder states
|
# decoder states
|
||||||
|
|
Loading…
Reference in New Issue