try reducing tolerance again

FluxML · Sep 26, 2024 · 9aeb7ba · 9aeb7ba
1 parent 1adf9cf
commit 9aeb7ba
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test/classifier.jl b/test/classifier.jl
@@ -110,7 +110,8 @@ end
 
 # check different resources (CPU1, CUDALibs, etc)) give about the same loss:
 reference = losses[1]
-@test all(x -> abs(x - reference) / reference < 1e-3, losses[2:end])
+println("losses for each resource: $losses")
+@test all(x -> abs(x - reference) / reference < 5e-3, losses[2:end])
 
 
 # # NEURAL NETWORK BINARY CLASSIFIER