fix concat problem when uttr is on cuda device

yistLin · May 20, 2021 · a7ac334 · a7ac334
1 parent 876394a
commit a7ac334
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/modules/dvector.py b/modules/dvector.py
@@ -55,9 +55,8 @@ def embed_utterance(self, utterance: Tensor) -> Tensor:
             # Pad to multiple of hop length
             hop_len = self.seg_len // 2
             tgt_len = math.ceil(utterance.size(0) / hop_len) * hop_len
-            padded = torch.cat(
-                [utterance, torch.zeros(tgt_len - utterance.size(0), utterance.size(1))]
-            )
+            zero_padding = torch.zeros(tgt_len - utterance.size(0), utterance.size(1))
+            padded = torch.cat([utterance, zero_padding.to(utterance.device)])
 
             segments = padded.unfold(0, self.seg_len, self.seg_len // 2)
             segments = segments.transpose(1, 2)  # (batch, seg_len, mel_dim)