Merge pull request #4941 from pyf98/fix

Fix device mismatch error in GPU decoding with PyTorch 1.13
espnet · Feb 15, 2023 · 9c7bde4 · 9c7bde4
2 parents 78c93bc + 6bf7987
commit 9c7bde4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/espnet/nets/batch_beam_search.py b/espnet/nets/batch_beam_search.py
@@ -343,5 +343,5 @@ def post_process(
         for b in torch.nonzero(is_eos, as_tuple=False).view(-1):
             hyp = self._select(running_hyps, b)
             ended_hyps.append(hyp)
-        remained_ids = torch.nonzero(is_eos == 0, as_tuple=False).view(-1)
+        remained_ids = torch.nonzero(is_eos == 0, as_tuple=False).view(-1).cpu()
         return self._batch_select(running_hyps, remained_ids)