From 81b6fbf19d115de02b53dc34afe23b4833fa45da Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Fri, 28 Oct 2022 18:19:06 +0000 Subject: [PATCH] higher precision for vae --- tests/models/test_models_vae.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/test_models_vae.py b/tests/models/test_models_vae.py index 95e5f656..f6333d6c 100644 --- a/tests/models/test_models_vae.py +++ b/tests/models/test_models_vae.py @@ -209,7 +209,7 @@ class AutoencoderKLIntegrationTests(unittest.TestCase): output_slice = sample[-1, -2:, :2, -2:].flatten().float().cpu() expected_output_slice = torch.tensor(expected_slice) - assert torch_all_close(output_slice, expected_output_slice, atol=5e-3) + assert torch_all_close(output_slice, expected_output_slice, atol=1e-2) @parameterized.expand( [