bug fix

2018-03-23 05:18:51 -07:00 · 2018-03-23 05:18:51 -07:00 · c61a4419e5
parent 46a2b10038
commit c61a4419e5
1 changed files with 1 additions and 1 deletions
--- a/layers/tacotron.py
+++ b/layers/tacotron.py
@ -253,7 +253,7 @@ class Decoder(nn.Module):
        B = inputs.size(0)

        # Run greedy decoding if memory is None
-        greedy = ~self.training
+        greedy = not self.training

        if memory is not None:
            # Grouping multiple frames if necessary