Commit 75ca501a authored by Nikita Shulga's avatar Nikita Shulga Committed by Facebook GitHub Bot
Browse files

Relax absolute tolerance for Kaldi compat tests (#2165)

Summary:
Find out that tests are failing after change for tester GPU class, see https://github.com/pytorch/audio/pull/1791

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

Reviewed By: mthrok

Differential Revision: D33674802

Pulled By: malfet

fbshipit-source-id: 2e39386c0f129cf44a30d5dfea67e9e2d0e875cf
parent 8d4e17a2
...@@ -40,7 +40,7 @@ class Kaldi(TempDirMixin, TestBaseMixin): ...@@ -40,7 +40,7 @@ class Kaldi(TempDirMixin, TestBaseMixin):
result = torchaudio.compliance.kaldi.spectrogram(waveform, **kwargs) result = torchaudio.compliance.kaldi.spectrogram(waveform, **kwargs)
command = ["compute-spectrogram-feats"] + convert_args(**kwargs) + ["scp:-", "ark:-"] command = ["compute-spectrogram-feats"] + convert_args(**kwargs) + ["scp:-", "ark:-"]
kaldi_result = run_kaldi(command, "scp", wave_file) kaldi_result = run_kaldi(command, "scp", wave_file)
self.assert_equal(result, expected=kaldi_result, rtol=1e-4, atol=1e-8) self.assert_equal(result, expected=kaldi_result, rtol=1e-4, atol=1e-6)
@parameterized.expand(load_params("kaldi_test_mfcc_args.jsonl")) @parameterized.expand(load_params("kaldi_test_mfcc_args.jsonl"))
@skipIfNoExec("compute-mfcc-feats") @skipIfNoExec("compute-mfcc-feats")
...@@ -51,4 +51,4 @@ class Kaldi(TempDirMixin, TestBaseMixin): ...@@ -51,4 +51,4 @@ class Kaldi(TempDirMixin, TestBaseMixin):
result = torchaudio.compliance.kaldi.mfcc(waveform, **kwargs) result = torchaudio.compliance.kaldi.mfcc(waveform, **kwargs)
command = ["compute-mfcc-feats"] + convert_args(**kwargs) + ["scp:-", "ark:-"] command = ["compute-mfcc-feats"] + convert_args(**kwargs) + ["scp:-", "ark:-"]
kaldi_result = run_kaldi(command, "scp", wave_file) kaldi_result = run_kaldi(command, "scp", wave_file)
self.assert_equal(result, expected=kaldi_result, rtol=1e-4, atol=1e-8) self.assert_equal(result, expected=kaldi_result, rtol=1e-4, atol=1e-5)
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