From f0d69c3f9e9c9adcdc329be850f1de357bf7b762 Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Mon, 29 Nov 2021 07:17:08 +0800 Subject: [PATCH] Fix chunk size (#22) * Use chunk size in feature extraction. * Fix a typo. --- MANIFEST.in | 2 ++ kaldifeat/python/kaldifeat/offline_feature.py | 4 ++-- setup.py | 1 - 3 files changed, 4 insertions(+), 3 deletions(-) diff --git a/MANIFEST.in b/MANIFEST.in index 6ee0933..afb672e 100644 --- a/MANIFEST.in +++ b/MANIFEST.in @@ -1,3 +1,5 @@ +include LICENSE +include README.md include CMakeLists.txt exclude pyproject.toml recursive-include kaldifeat *.* diff --git a/kaldifeat/python/kaldifeat/offline_feature.py b/kaldifeat/python/kaldifeat/offline_feature.py index 0d2a8c3..68948a0 100644 --- a/kaldifeat/python/kaldifeat/offline_feature.py +++ b/kaldifeat/python/kaldifeat/offline_feature.py @@ -72,7 +72,7 @@ class OfflineFeature(nn.Module): strided = [self.convert_samples_to_frames(w) for w in waves] strided = torch.cat(strided, dim=0) - features = self.compute(strided, vtln_warp) + features = self.compute(strided, vtln_warp, chunk_size=chunk_size) if is_list: return list(features.split(num_frames_per_wave)) @@ -122,7 +122,7 @@ class OfflineFeature(nn.Module): ) features.append(this_chunk) if end < x.size(0): - last_chunk = self.compute_features(x[end:], vtln_warp) + last_chunk = self.computer.compute_features(x[end:], vtln_warp) features.append(last_chunk) features = torch.cat(features, dim=0) diff --git a/setup.py b/setup.py index 4f45bfc..57250c4 100644 --- a/setup.py +++ b/setup.py @@ -34,7 +34,6 @@ setuptools.setup( version=get_package_version(), author="Fangjun Kuang", author_email="csukuangfj@gmail.com", - data_files=[("", ["LICENSE", "README.md"])], package_dir={package_name: "kaldifeat/python/kaldifeat"}, packages=[package_name], url="https://github.com/csukuangfj/kaldifeat",