diff --git a/tests/test_training.py b/tests/test_training.py
index 73611e81a279ed911110d3e8c1c7308e90fb53df..f146afbaddad0dafddf9ec4a76d6ce983ae58c6c 100644
--- a/tests/test_training.py
+++ b/tests/test_training.py
@@ -99,7 +99,7 @@ def test_train_and_test(
                     assert torch.allclose(
                         trained_tensor,
                         expected_tensor,
-                        atol=1e-04,
+                        atol=1e-03,
                     )
             assert (
                 trained_model[optimizer_part]["param_groups"]