From ea22a4119b398be23026dc60044ac836a2db09b4 Mon Sep 17 00:00:00 2001 From: krishnakalyan3 Date: Wed, 21 Apr 2021 02:33:29 +0200 Subject: [PATCH 1/4] melscale unit test --- test/torchaudio_unittest/transforms/autograd_test_impl.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/test/torchaudio_unittest/transforms/autograd_test_impl.py b/test/torchaudio_unittest/transforms/autograd_test_impl.py index 7c368445a8..91a7c75929 100644 --- a/test/torchaudio_unittest/transforms/autograd_test_impl.py +++ b/test/torchaudio_unittest/transforms/autograd_test_impl.py @@ -137,6 +137,12 @@ def test_amplitude_to_db(self): waveform = get_whitenoise(sample_rate=sample_rate, duration=0.05, n_channels=2) self.assert_grad(transform, [waveform]) + def test_melscale(self): + sample_rate = 8000 + transform = T.MelScale(sample_rate=sample_rate) + waveform = get_whitenoise(sample_rate=sample_rate, duration=0.05, n_channels=2) + self.assert_grad(transform, [waveform]) + @parameterized.expand([(1.5, "amplitude"), (2, "power"), (10, "db")]) def test_vol(self, gain, gain_type): sample_rate = 8000 From 609d74fd33158af92a8db6ce04bd2d01f5cf0032 Mon Sep 17 00:00:00 2001 From: krishnakalyan3 Date: Thu, 22 Apr 2021 07:53:26 +0200 Subject: [PATCH 2/4] fix dims --- test/torchaudio_unittest/transforms/autograd_test_impl.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/test/torchaudio_unittest/transforms/autograd_test_impl.py b/test/torchaudio_unittest/transforms/autograd_test_impl.py index 91a7c75929..8fb5f9a725 100644 --- a/test/torchaudio_unittest/transforms/autograd_test_impl.py +++ b/test/torchaudio_unittest/transforms/autograd_test_impl.py @@ -139,8 +139,11 @@ def test_amplitude_to_db(self): def test_melscale(self): sample_rate = 8000 + n_fft = 400 transform = T.MelScale(sample_rate=sample_rate) - waveform = get_whitenoise(sample_rate=sample_rate, duration=0.05, n_channels=2) + waveform = get_spectrogram( + get_whitenoise(sample_rate=sample_rate, duration=0.05, n_channels=2), + n_fft=n_fft, power=1) self.assert_grad(transform, [waveform]) @parameterized.expand([(1.5, "amplitude"), (2, "power"), (10, "db")]) From afaa6320d496a4e51f3f4c085804f47785624f42 Mon Sep 17 00:00:00 2001 From: krishnakalyan3 Date: Thu, 22 Apr 2021 07:54:44 +0200 Subject: [PATCH 3/4] add missing params --- test/torchaudio_unittest/transforms/autograd_test_impl.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/test/torchaudio_unittest/transforms/autograd_test_impl.py b/test/torchaudio_unittest/transforms/autograd_test_impl.py index 8fb5f9a725..35889fbf1f 100644 --- a/test/torchaudio_unittest/transforms/autograd_test_impl.py +++ b/test/torchaudio_unittest/transforms/autograd_test_impl.py @@ -140,7 +140,8 @@ def test_amplitude_to_db(self): def test_melscale(self): sample_rate = 8000 n_fft = 400 - transform = T.MelScale(sample_rate=sample_rate) + n_mels = n_fft // 2 + 1 + transform = T.MelScale(sample_rate=sample_rate, n_mels=n_mels) waveform = get_spectrogram( get_whitenoise(sample_rate=sample_rate, duration=0.05, n_channels=2), n_fft=n_fft, power=1) From a5aa15af368e72bdb7d23ef8f41d85bfd6d22460 Mon Sep 17 00:00:00 2001 From: krishnakalyan3 Date: Fri, 23 Apr 2021 07:11:02 +0200 Subject: [PATCH 4/4] update var for readbility --- test/torchaudio_unittest/transforms/autograd_test_impl.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/torchaudio_unittest/transforms/autograd_test_impl.py b/test/torchaudio_unittest/transforms/autograd_test_impl.py index 35889fbf1f..b8fe6c1863 100644 --- a/test/torchaudio_unittest/transforms/autograd_test_impl.py +++ b/test/torchaudio_unittest/transforms/autograd_test_impl.py @@ -142,10 +142,10 @@ def test_melscale(self): n_fft = 400 n_mels = n_fft // 2 + 1 transform = T.MelScale(sample_rate=sample_rate, n_mels=n_mels) - waveform = get_spectrogram( + spec = get_spectrogram( get_whitenoise(sample_rate=sample_rate, duration=0.05, n_channels=2), n_fft=n_fft, power=1) - self.assert_grad(transform, [waveform]) + self.assert_grad(transform, [spec]) @parameterized.expand([(1.5, "amplitude"), (2, "power"), (10, "db")]) def test_vol(self, gain, gain_type):