Autograd tests for Transforms MFCC (#1415)

Co-authored-by: krishnakalyan3 <[email protected]>
pytorch · Apr 2, 2021 · 404fa12 · 404fa12
1 parent 52decd2
commit 404fa12
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/test/torchaudio_unittest/transforms/autograd_test_impl.py b/test/torchaudio_unittest/transforms/autograd_test_impl.py
@@ -60,3 +60,10 @@ def test_melspectrogram(self):
         transform = T.MelSpectrogram(sample_rate=sample_rate)
         waveform = get_whitenoise(sample_rate=sample_rate, duration=0.05, n_channels=2)
         self.assert_grad(transform, [waveform], nondet_tol=1e-10)
+
+    @parameterized.expand([(False, ), (True, )])
+    def test_mfcc(self, log_mels):
+        sample_rate = 8000
+        transform = T.MFCC(sample_rate=sample_rate, log_mels=log_mels)
+        waveform = get_whitenoise(sample_rate=sample_rate, duration=0.05, n_channels=2)
+        self.assert_grad(transform, [waveform])