Skip to content

Commit 817f2dc

Browse files
committed
Widen AlphaDropout test tolerance
1 parent 16489b3 commit 817f2dc

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

test/layers/normalisation.jl

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -78,16 +78,16 @@ end
7878

7979
y = evalwgrad(m, x)
8080
# Should preserve unit mean and variance
81-
@test mean(y) 0 atol=0.1
82-
@test var(y) 1 atol=0.1
81+
@test mean(y) 0 atol=0.2
82+
@test var(y) 1 atol=0.2
8383

8484
testmode!(m, true) # should override istraining
8585
@test evalwgrad(m, x) == x
8686

8787
testmode!(m, false)
8888
y = evalwgrad(m, x)
89-
@test mean(y) 0 atol=0.1
90-
@test var(y) 1 atol=0.1
89+
@test mean(y) 0 atol=0.2
90+
@test var(y) 1 atol=0.2
9191

9292
# Known good value ranges
9393
# Values taken from https://github.com/pytorch/pytorch/blob/v1.10.0/test/cpp/api/modules.cpp#L1337-L1338

0 commit comments

Comments
 (0)