tests bias switch
This commit is contained in:
parent
040697fb2b
commit
b596faaffa
@ -28,9 +28,25 @@ end
|
|||||||
op = bias(ip)
|
op = bias(ip)
|
||||||
@test sum(op) == prod(size(op))
|
@test sum(op) == prod(size(op))
|
||||||
|
|
||||||
bias = Conv((2,2), 1=>3, bias = zero(3))
|
bias = Conv((2,2), 1=>3, bias = Flux.ZeroType((3,)))
|
||||||
op = bias(ip)
|
op = bias(ip)
|
||||||
@test sum(op) === 0.f0
|
@test sum(op) === 0.f0
|
||||||
|
|
||||||
|
# Train w/o bias and make sure no convergence happens
|
||||||
|
# when only bias can be converged
|
||||||
|
bias = Conv((2, 2), 1=>3, bias = Flux.ZeroType((3,)));
|
||||||
|
ip = zeros(Float32, 28,28,1,1)
|
||||||
|
op = zeros(Float32, 27,27,3,1) .+ 2.f0
|
||||||
|
opt = Descent()
|
||||||
|
|
||||||
|
for _ = 1:10^3
|
||||||
|
gs = gradient(params(bias)) do
|
||||||
|
Flux.mse(bias(ip), op)
|
||||||
|
end
|
||||||
|
Flux.Optimise.update!(opt, params(bias), gs)
|
||||||
|
end
|
||||||
|
|
||||||
|
@test Flux.mse(bias(ip), op) ≈ 4.f0
|
||||||
end
|
end
|
||||||
|
|
||||||
@testset "asymmetric padding" begin
|
@testset "asymmetric padding" begin
|
||||||
|
Loading…
Reference in New Issue
Block a user