small bug fixes

2026-09-11 12:00:24 +08:00 · 2019-05-14 13:53:26 +02:00
parent 2b60f9a731
commit bb2b705e01
2 changed files with 3 additions and 0 deletions
@@ -509,6 +509,7 @@ class Decoder(nn.Module):
        else:
            self._init_states(inputs, mask=None, keep_states=True)

+        self.attention_layer.init_win_idx()
        self.attention_layer.init_states(inputs)
        outputs, stop_tokens, alignments, t = [], [], [], 0
        stop_flags = [False, False, False]