From a474a57456c3df02d5d8e3e7b0e2db2c36df40e2 Mon Sep 17 00:00:00 2001 From: Matthew Baas Date: Tue, 30 Nov 2021 16:04:24 +0200 Subject: [PATCH] small bugfix --- ge2e_embedding.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ge2e_embedding.py b/ge2e_embedding.py index 652494a..b0fa096 100644 --- a/ge2e_embedding.py +++ b/ge2e_embedding.py @@ -26,9 +26,9 @@ def melspec_from_file(self, x): """ Convert file path `x` to a [n_frames, n_mels] shape mel-spectrogram tensor """ return self.melspec_tfm.from_file(x).T - def melspec_from_array(self, x): + def melspec_from_array(self, x, sr): """ Convert a 1D torch tensor in to a [n_frames, n_mels] shape mel-spectrogram tensor """ - return self.melspec_tfm.from_array(x).T + return self.melspec_tfm.from_array(x, sr).T def print_hparams(self): for key in self.hparams.__dict__():