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

fix stateful in backends

Browse Source
This commit is contained in:
Mike J Innes 2017-06-02 16:28:31 +01:00
parent 5232a3300c
commit 10abb64f4b
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/backend/mxnet/model.jl
View File

@ -116,7 +116,7 @@ Flux.update!(m::Model, η) = (update!(m.last, η); m)
using Flux: Stateful, SeqModel
mxnet(m::Stateful) = Stateful(mxnet(m.model), m.istate, m.ostate)
mxnet(m::Stateful) = Stateful(mxnet(m.model), m.states, m.istate, m.ostate)
mxnet(m::SeqModel) = SeqModel(mxnet(m.model), m.steps)
# MX FeedForward interface

2
src/backend/tensorflow/model.jl
View File

@ -77,5 +77,5 @@ Flux.update!(m::Model, η) = (update!(m.exec, η); m)
using Flux: Stateful, SeqModel
tf(m::Stateful) = Stateful(tf(m.model), m.istate, m.ostate)
tf(m::Stateful) = Stateful(tf(m.model), m.states, m.istate, m.ostate)
tf(m::SeqModel) = SeqModel(tf(m.model), m.steps)