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

actually use init argument in LSTMCell

Browse Source
This commit is contained in:
Dave Kleinschmidt 2017-10-03 14:16:51 -04:00 committed by Mike J Innes
parent c202e2bc1a
commit 2b95aff158
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/layers/recurrent.jl
View File

@ -63,9 +63,9 @@ struct LSTMCell{D1,D2,V}
end
function LSTMCell(in, out; init = initn)
cell = LSTMCell([Dense(in+out, out, σ, init = initn) for _ = 1:3]...,
Dense(in+out, out, tanh, init = initn),
param(initn(out)), param(initn(out)))
cell = LSTMCell([Dense(in+out, out, σ, init = init) for _ = 1:3]...,
Dense(in+out, out, tanh, init = init),
param(init(out)), param(init(out)))
cell.forget.b.data .= 1
return cell
end