diff --git a/layers/tacotron.py b/layers/tacotron.py index 51548287..983855d4 100644 --- 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