Fix unit tests for griffinlim and Spectrogram (#3099)

Summary: Pull Request resolved: https://github.com/pytorch/audio/pull/3099

Reviewed By: mthrok

Differential Revision: D43596866

Pulled By: nateanl

fbshipit-source-id: 43a139bf8ebdf3261414e2855aefc3b53df298ac

test/torchaudio_unittest/functional/batch_consistency_test.py

@@ -65,7 +65,7 @@ class TestFunctional(common_utils.TorchaudioTestCase):
             "rand_init": False,
         }
         func = partial(F.griffinlim, **kwargs)
-        self.assert_batch_consistency(func, inputs=(batch,), atol=5e-5)
+        self.assert_batch_consistency(func, inputs=(batch,), atol=1e-4)
 
     @parameterized.expand(
         list(

test/torchaudio_unittest/transforms/librosa_compatibility_test_impl.py

@@ -36,7 +36,7 @@ class TransformsTestBase(TestBaseMixin):
         result = T.Spectrogram(n_fft=n_fft, hop_length=hop_length, power=power,).to(self.device, self.dtype)(
             waveform
         )[0]
-        self.assertEqual(result, torch.from_numpy(expected), atol=1e-5, rtol=1e-5)
+        self.assertEqual(result, torch.from_numpy(expected), atol=1e-4, rtol=1e-4)
 
     def test_Spectrogram_complex(self):
         n_fft = 400
@@ -54,7 +54,7 @@ class TransformsTestBase(TestBaseMixin):
         result = T.Spectrogram(n_fft=n_fft, hop_length=hop_length, power=None, return_complex=True,).to(
             self.device, self.dtype
         )(waveform)[0]
-        self.assertEqual(result.abs(), torch.from_numpy(expected), atol=1e-5, rtol=1e-5)
+        self.assertEqual(result.abs(), torch.from_numpy(expected), atol=1e-4, rtol=1e-4)
 
     @nested_params(
         [