Increase test_core.py tolerance

sarahshi · May 28, 2024 · acdcb23 · acdcb23
1 parent 90fb446
commit acdcb23
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/UnitTests/test_core.py b/UnitTests/test_core.py
@@ -116,7 +116,7 @@ def __init__(self):
             self.bn1 = nn.BatchNorm2d(20)
 
 
-    def is_normal_distribution(self, tensor, mean, std, tolerance=0.01):
+    def is_normal_distribution(self, tensor, mean, std, tolerance=0.05):
         # Check if the mean of the tensor is close to the expected mean
         mean_close = math.isclose(tensor.mean().item(), mean, abs_tol=tolerance)
         # Check if the standard deviation of the tensor is close to the expected std