BatchNorm test corrected
This commit is contained in:
parent
f465665c73
commit
a56cfb73c3
@ -37,7 +37,7 @@ end
|
||||
# initial m.μ is 0
|
||||
|
||||
y = trainmode(m, x)
|
||||
@test y ≈ [-1.22474 0 1.22474; -1.22474 0 1.22474]
|
||||
@test isapprox(y, [-1.22474 0 1.22474; -1.22474 0 1.22474], atol = 1.0e-5)
|
||||
# julia> x
|
||||
# 2×3 Array{Float64,2}:
|
||||
# 1.0 3.0 5.0
|
||||
|
Loading…
Reference in New Issue
Block a user