Unverified Commit 404fa12a authored by Krishna Kalyan's avatar Krishna Kalyan Committed by GitHub
Browse files

Autograd tests for Transforms MFCC (#1415)


Co-authored-by: default avatarkrishnakalyan3 <skalyan@cloudera.com>
parent 52decd2a
...@@ -60,3 +60,10 @@ class AutogradTestMixin(TestBaseMixin): ...@@ -60,3 +60,10 @@ class AutogradTestMixin(TestBaseMixin):
transform = T.MelSpectrogram(sample_rate=sample_rate) transform = T.MelSpectrogram(sample_rate=sample_rate)
waveform = get_whitenoise(sample_rate=sample_rate, duration=0.05, n_channels=2) waveform = get_whitenoise(sample_rate=sample_rate, duration=0.05, n_channels=2)
self.assert_grad(transform, [waveform], nondet_tol=1e-10) 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])
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment