bug fix

2026-09-10 11:50:20 +08:00 · 2018-03-23 05:18:51 -07:00
parent 2617518d91
commit df4a644326
1 changed files with 1 additions and 1 deletions
@@ -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