bug fix

2026-09-11 12:00:24 +08:00 · 2018-05-11 04:19:28 -07:00
parent 31cc9c5443
commit 8ed9f57a6d
1 changed files with 1 additions and 1 deletions
@@ -303,7 +303,7 @@ class Decoder(nn.Module):
            stop_token = self.stopnet(output)
            outputs += [output]
            alignments += [alignment]
-            stop_tokens += stop_token
+            stop_tokens += [stop_token]
            t += 1
            if (not greedy and self.training) or (greedy and memory is not None):
                if t >= T_decoder: