diff --git a/lhotse/features/kaldi/extractors.py b/lhotse/features/kaldi/extractors.py index 0057f4d49..d40db0322 100644 --- a/lhotse/features/kaldi/extractors.py +++ b/lhotse/features/kaldi/extractors.py @@ -309,7 +309,7 @@ def frame_shift(self) -> Seconds: return self.config.frame_shift def feature_dim(self, sampling_rate: int) -> int: - return self.config.num_ceps + return self.extractor.fft_length // 2 + 1 def extract( self, samples: Union[np.ndarray, torch.Tensor], sampling_rate: int @@ -415,7 +415,7 @@ def frame_shift(self) -> Seconds: return self.config.frame_shift def feature_dim(self, sampling_rate: int) -> int: - return self.config.num_ceps + return self.extractor.fft_length // 2 + 1 def extract( self, samples: Union[np.ndarray, torch.Tensor], sampling_rate: int