From cf72c0a3b2f1981390b6c13aab69e61ac76b271e Mon Sep 17 00:00:00 2001 From: Ping Zheng Date: Wed, 15 Nov 2023 12:12:48 -0500 Subject: [PATCH] fix --- i6_models/primitives/feature_extraction.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/i6_models/primitives/feature_extraction.py b/i6_models/primitives/feature_extraction.py index 824267c3..ca716b37 100644 --- a/i6_models/primitives/feature_extraction.py +++ b/i6_models/primitives/feature_extraction.py @@ -123,7 +123,7 @@ def forward(self, raw_audio, length) -> Tuple[torch.Tensor, torch.Tensor]: log_melspec = torch.log10(torch.clamp(melspec, min=self.min_amp)) feature_data = torch.transpose(log_melspec, 1, 2) # [B, T', F'] - if self.center and not rasr_compatible: + if self.center and not self.rasr_compatible: length = (length // self.hop_length) + 1 else: length = ((length - self.win_length) // self.hop_length) + 1