Skip to content

GitLab

Unverified Commit 931555c1 authored by Krishna Kalyan's avatar Krishna Kalyan Committed by GitHub
Browse files

Add autograd test for T.AmplitudeToDB (#1447)

parent e911e5eb
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 0 deletions
test/torchaudio_unittest/transforms/autograd_test_impl.py
View file @ 931555c1
...@@ -123,3 +123,9 @@ class AutogradTestMixin(TestBaseMixin): ...@@ -123,3 +123,9 @@ class AutogradTestMixin(TestBaseMixin):
transform = T.SpectralCentroid(sample_rate=sample_rate) transform = T.SpectralCentroid(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)
def test_amplitude_to_db(self):
sample_rate = 8000
transform = T.AmplitudeToDB()
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