small fix

rwth-i6 · Jan 15, 2025 · e0d4767 · e0d4767
1 parent 3fa3ca2
commit e0d4767
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/users/zeyer/experiments/exp2024_04_23_baselines/recog_ext/ctc_label_sync_espnet.py b/users/zeyer/experiments/exp2024_04_23_baselines/recog_ext/ctc_label_sync_espnet.py
@@ -324,7 +324,8 @@ def batch_score(self, ys: torch.Tensor, states: List[Any], xs: torch.Tensor) ->
                 def _batchify(*args):
                     assert len(args) == batch_size
                     if isinstance(args[0], Tensor):
-                        return rf.stack(args, out_dim=batch_dim_)
+                        x, _ = rf.stack(args, out_dim=batch_dim_)
+                        return x
                     if isinstance(args[0], Dim):
                         assert all(args[0] == s for s in args)
                         return args[0]