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

4
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
@ -57,7 +57,7 @@ end
# 1.3 # 1.3
# 1.3 # 1.3
@test m.σ² ≈ .1 .* var(x, dims = 2, corrected=false) .* (3 / 2).+ .9 .* [1., 1.] @test m.σ² ≈ .1 .* var(x, dims = 2, corrected=false) .* (3 / 2).+ .9 .* [1., 1.]
x′ = m(x) x′ = m(x)
@test isapprox(x′[1], (1 .- 0.3) / sqrt(1.3), atol = 1.0e-5) @test isapprox(x′[1], (1 .- 0.3) / sqrt(1.3), atol = 1.0e-5)
end end