From 4cbbaf706e9b56bd9b5f0817b2ea2e50c35f11ad Mon Sep 17 00:00:00 2001 From: nateanl Date: Mon, 1 Nov 2021 16:59:19 +0000 Subject: [PATCH] add spec setting for mfcc --- examples/hubert/utils/feature_utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/examples/hubert/utils/feature_utils.py b/examples/hubert/utils/feature_utils.py index eb76031082..26078065b5 100644 --- a/examples/hubert/utils/feature_utils.py +++ b/examples/hubert/utils/feature_utils.py @@ -68,7 +68,9 @@ def extract_feature( waveform = waveform[0].to(device) if feature_type == "mfcc": feature_extractor = torchaudio.transforms.MFCC( - sample_rate=sample_rate + sample_rate=sample_rate, + n_mfcc=13, + melkwargs={'n_fft': 400, 'hop_length': 160, 'center': False} ).to(device) mfccs = feature_extractor(waveform) # (freq, time) # mfccs = torchaudio.compliance.kaldi.mfcc(