Use correct signature for torch.tensor.type() (from Samsung, AI Center, Cambridge) (speechbrain#2910)

rogiervd · Rogier van Dalen · pplantinga · commit 94caa907c9b3 · 2025-06-02T14:12:29.000-04:00
Co-authored-by: Rogier van Dalen &lt;r.vandalen@samsung.com&gt;
diff --git a/speechbrain/alignment/aligner.py b/speechbrain/alignment/aligner.py
@@ -739,7 +739,7 @@ def _dp_viterbi(
             )
             v_matrix[:, :, t] = x + emiss_pred_useful[:, :, t]
 
-            backpointers[:, :, t] = argmax.type(torch.FloatTensor)
+            backpointers[:, :, t] = argmax.type(dtype=torch.float32)
 
         z_stars = []
         z_stars_loc = []

Original file line number	Diff line number	Diff line change
`@@ -739,7 +739,7 @@ def _dp_viterbi(`
`739`	`739`	`)`
`740`	`740`	`v_matrix[:, :, t] = x + emiss_pred_useful[:, :, t]`
`741`	`741`
`742`		`- backpointers[:, :, t] = argmax.type(torch.FloatTensor)`
	`742`	`+ backpointers[:, :, t] = argmax.type(dtype=torch.float32)`
`743`	`743`
`744`	`744`	`z_stars = []`
`745`	`745`	`z_stars_loc = []`