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

update for rnn wrappers

Browse Source
This commit is contained in:
Mike J Innes 2017-04-27 17:27:08 +01:00
parent f5f5e77239
commit 121af0579a
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/model.jl
View File

@ -146,6 +146,8 @@ function back!(m::Stateful, Δ, x)
back!(m.model, ((zeros.(m.ostate)...,), Δ), (m.istate...,), x)[2:end] back!(m.model, ((zeros.(m.ostate)...,), Δ), (m.istate...,), x)[2:end]
end end
update!(m::Stateful, η) = update!(m.model, η)
stateless(m) = m stateless(m) = m
stateless(m::Stateful) = m.model stateless(m::Stateful) = m.model
@ -166,3 +168,5 @@ function (m::SeqModel)(x)
end end
back!(m::SeqModel, Δ, x) = (runseq((Δ, x) -> back!(m.model, Δ, x)[1], Δ, x),) back!(m::SeqModel, Δ, x) = (runseq((Δ, x) -> back!(m.model, Δ, x)[1], Δ, x),)
update!(m::SeqModel, η) = update!(m.model, η)