disable test comparing losses b/w CPU/GPU in presence of dropout

FluxML · May 1, 2024 · f60be3f · f60be3f
1 parent 99100b8
commit f60be3f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/image.jl b/test/image.jl
@@ -74,7 +74,7 @@ end
 # check different resources (CPU1, CUDALibs) give about the same loss:
 reference = losses[1]
 @info "Losses for each computational resource: $losses"
-@test all(x->abs(x - reference)/reference < 5e-4, losses[2:end])
+# @test_broken all(x->abs(x - reference)/reference < 5e-4, losses[2:end])
 
 
 # # BASIC IMAGE TESTS COLOR
@@ -118,7 +118,7 @@ end
 # check different resources (CPU1, CUDALibs, etc)) give about the same loss:
 reference = losses[1]
 @info "Losses for each computational resource: $losses"
-@test all(x->abs(x - reference)/reference < 1e-5, losses[2:end])
+# @test_broken all(x->abs(x - reference)/reference < 1e-5, losses[2:end])
 
 
 # # SMOKE TEST FOR DEFAULT BUILDER