From 4544a4e9f1d09af7797575f2e31156897f3dddab Mon Sep 17 00:00:00 2001 From: EmreOzkose Date: Thu, 4 Nov 2021 10:09:13 +0300 Subject: [PATCH] fix tensor device type incompatibility --- kaldifeat/python/kaldifeat/offline_feature.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/kaldifeat/python/kaldifeat/offline_feature.py b/kaldifeat/python/kaldifeat/offline_feature.py index 0d2a8c3..dc07ee0 100644 --- a/kaldifeat/python/kaldifeat/offline_feature.py +++ b/kaldifeat/python/kaldifeat/offline_feature.py @@ -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)