fix initialisation

This commit is contained in:
Mike J Innes 2018-01-10 14:11:52 +00:00
parent b44237468e
commit 8f8589a7f4

View File

@ -160,10 +160,11 @@ struct GRUCell{D1,D2,V}
h::V h::V
end end
function GRUCell(in, out; init = initn) function GRUCell(in, out)
cell = GRUCell([Dense(in+out, out, σ, init = init) for _ = 1:2]..., cell = GRUCell(Dense(in+out, out, σ),
Dense(in+out, out, tanh, init = init), Dense(in+out, out, σ),
param(init(out))) Dense(in+out, out, tanh),
param(initn(out)))
return cell return cell
end end