fix contiguous issue with dec_bos_cut

isi-nlp · Aug 21, 2020 · 0478060 · 0478060
1 parent 19c56bb
commit 0478060
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rtg/module/decoder.py b/rtg/module/decoder.py
@@ -346,7 +346,7 @@ def beam_decode(self, x_seqs, x_lens, max_len, beam_size=default_beam_size, num_
             if actives.sum() == 0:  # all sequences Ended
                 break
             # [Batch x Beams x Vocab] <-- [Batch x Beams x Time]
-            flat_ys = ys.view(batch_size * beam_size, -1)
+            flat_ys = ys.contiguous().view(batch_size * beam_size, -1)
             log_prob = gen.generate_next(flat_ys)  # ys upto current time step
             log_prob = log_prob.view(batch_size, beam_size, -1)