Fix chunk size (#22)

* Use chunk size in feature extraction.

* Fix a typo.
This commit is contained in:
Fangjun Kuang 2021-11-29 07:17:08 +08:00 committed by GitHub
parent 632098e0c1
commit f0d69c3f9e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 4 additions and 3 deletions

View File

@ -1,3 +1,5 @@
include LICENSE
include README.md
include CMakeLists.txt
exclude pyproject.toml
recursive-include kaldifeat *.*

View File

@ -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)

View File

@ -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",