[FIX] Increase tol

f-dangel · Jul 2, 2023 · 6ee055b · 6ee055b
1 parent 2ccce51
commit 6ee055b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/tutorials/basic_conv2d.py b/docs/tutorials/basic_conv2d.py
@@ -25,4 +25,4 @@
 ein_layer.weight.data = torch_layer.weight.data
 ein_layer.bias.data = torch_layer.bias.data
 
-assert allclose(torch_layer(x), ein_layer(x), atol=1e-7)
+assert allclose(torch_layer(x), ein_layer(x), rtol=1e-4, atol=1e-6)