Unverified Commit 90311ed1 authored by Krishna Kalyan's avatar Krishna Kalyan Committed by GitHub
Browse files

Add Autograd test for T.Resample (#1416)

parent 6a677ac8
...@@ -71,3 +71,9 @@ class AutogradTestMixin(TestBaseMixin): ...@@ -71,3 +71,9 @@ class AutogradTestMixin(TestBaseMixin):
transform = T.MFCC(sample_rate=sample_rate, log_mels=log_mels) transform = T.MFCC(sample_rate=sample_rate, log_mels=log_mels)
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]) self.assert_grad(transform, [waveform])
@parameterized.expand([(8000, 8000), (8000, 4000), (4000, 8000)])
def test_resample(self, orig_freq, new_freq):
transform = T.Resample(orig_freq=orig_freq, new_freq=new_freq)
waveform = get_whitenoise(sample_rate=8000, 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