mirror of https://github.com/coqui-ai/TTS.git
bug fix
This commit is contained in:
parent
2617518d91
commit
df4a644326
|
@ -253,7 +253,7 @@ class Decoder(nn.Module):
|
||||||
B = inputs.size(0)
|
B = inputs.size(0)
|
||||||
|
|
||||||
# Run greedy decoding if memory is None
|
# Run greedy decoding if memory is None
|
||||||
greedy = ~self.training
|
greedy = not self.training
|
||||||
|
|
||||||
if memory is not None:
|
if memory is not None:
|
||||||
# Grouping multiple frames if necessary
|
# Grouping multiple frames if necessary
|
||||||
|
|
Loading…
Reference in New Issue