Add autograd test for T.SpectralCentroid (#1425)

c6eeebeb · Krishna Kalyan · GitHub · 8d2eeb11 · c6eeebeb
Unverified Commit c6eeebeb authored Apr 05, 2021 by Krishna Kalyan Committed by GitHub Apr 05, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

test/torchaudio_unittest/transforms/autograd_test_impl.py test/torchaudio_unittest/transforms/autograd_test_impl.py +6 -0

No files found.
--- a/test/torchaudio_unittest/transforms/autograd_test_impl.py
+++ b/test/torchaudio_unittest/transforms/autograd_test_impl.py
@@ -88,3 +88,9 @@ class AutogradTestMixin(TestBaseMixin):
        transform = T.Fade(fade_shape=fade_shape)
        waveform = get_whitenoise(sample_rate=8000, duration=0.05, n_channels=2)
        self.assert_grad(transform, [waveform], nondet_tol=1e-10)
+    def test_spectral_centroid(self):
+        sample_rate = 8000
+        transform = T.SpectralCentroid(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)