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

don't return hidden states

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

2
src/compiler/loops.jl
View File

@ -140,7 +140,7 @@ function stateless(s::Stateful)
v = graph(s.model) v = graph(s.model)
v = spliceinputs(v, group(constant.(s.states)...), v = spliceinputs(v, group(constant.(s.states)...),
[inputnode(i) for i = 1:graphinputs(v)-1]...) [inputnode(i) for i = 1:graphinputs(v)-1]...)
Capacitor(v) Capacitor(v[2])
end end
stateless(s::SeqModel) = SeqModel(stateless(s.model), s.steps) stateless(s::SeqModel) = SeqModel(stateless(s.model), s.steps)