Merge pull request espnet#5974 from YuzhuWang-code/fix_bug_gradient_t…

…runcation_in_pit_solver fix: gradient truncation bug in pit_solver.py
Masao-Someki · Dec 8, 2024 · 9e0876d · 9e0876d
2 parents efd0d59 + d5382f1
commit 9e0876d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/espnet2/enh/loss/wrappers/pit_solver.py b/espnet2/enh/loss/wrappers/pit_solver.py
@@ -98,9 +98,9 @@ def pair_loss(permutation):
                     perm0 = perm_.unsqueeze(1)
                 stats[k] = new_v.gather(1, perm0.to(device=new_v.device)).unbind(1)
         else:
-            loss = torch.tensor(
+            loss = torch.stack(
                 [
-                    torch.tensor(
+                    torch.stack(
                         [
                             pre_hook(
                                 self.criterion,