mirror of https://github.com/coqui-ai/TTS.git
bug fix
This commit is contained in:
parent
6e9360c455
commit
a25ba81296
|
@ -200,7 +200,7 @@ class Decoder(nn.Module):
|
||||||
|
|
||||||
def __init__(self, in_features, memory_dim, r):
|
def __init__(self, in_features, memory_dim, r):
|
||||||
super(Decoder, self).__init__()
|
super(Decoder, self).__init__()
|
||||||
self.mode = mode
|
self.r = r
|
||||||
self.max_decoder_steps = 200
|
self.max_decoder_steps = 200
|
||||||
self.memory_dim = memory_dim
|
self.memory_dim = memory_dim
|
||||||
# memory -> |Prenet| -> processed_memory
|
# memory -> |Prenet| -> processed_memory
|
||||||
|
|
Loading…
Reference in New Issue