mirrors/kaldifeat
1
0
Fork 0
mirror of https://github.com/csukuangfj/kaldifeat.git synced 2025-08-17 13:12:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix tensor device type incompatibility

Browse Source
This commit is contained in:
EmreOzkose 2021-11-04 10:09:13 +03:00
parent b3a90b1dcf
commit 4544a4e9f1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
kaldifeat/python/kaldifeat/offline_feature.py
View File

@ -69,7 +69,7 @@ class OfflineFeature(nn.Module):
for w in waves
]
strided = [self.convert_samples_to_frames(w) for w in waves]
strided = [self.convert_samples_to_frames(w.to(self.opts.device)) for w in waves]
strided = torch.cat(strided, dim=0)
features = self.compute(strided, vtln_warp)