TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

BatchNorm test corrected

Browse Source
This commit is contained in:
thebhatman 2019-06-11 20:34:48 +05:30
parent f465665c73
commit a56cfb73c3
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/layers/normalisation.jl
View File

@ -37,7 +37,7 @@ end
# initial m.μ is 0 # initial m.μ is 0
y = trainmode(m, x) 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 # julia> x
# 2×3 Array{Float64,2}: # 2×3 Array{Float64,2}:
# 1.0 3.0 5.0 # 1.0 3.0 5.0