Skip to content

GitLab

Commit 81b6fbf1 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

higher precision for vae

parent a7ae808e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/models/test_models_vae.py
View file @ 81b6fbf1
...@@ -209,7 +209,7 @@ class AutoencoderKLIntegrationTests(unittest.TestCase): ...@@ -209,7 +209,7 @@ class AutoencoderKLIntegrationTests(unittest.TestCase):
output_slice = sample[-1, -2:, :2, -2:].flatten().float().cpu() output_slice = sample[-1, -2:, :2, -2:].flatten().float().cpu()
expected_output_slice = torch.tensor(expected_slice) 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( @parameterized.expand(
[ [
......
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