don't return hidden states

2017-06-02 16:22:55 +01:00 · 2017-06-02 16:22:55 +01:00 · 5232a3300c
commit 5232a3300c
parent baaf9f2682
1 changed files with 1 additions and 1 deletions
--- a/src/compiler/loops.jl
+++ b/src/compiler/loops.jl
@ -140,7 +140,7 @@ function stateless(s::Stateful)
  v = graph(s.model)
  v = spliceinputs(v, group(constant.(s.states)...),
                   [inputnode(i) for i = 1:graphinputs(v)-1]...)
-  Capacitor(v)
+  Capacitor(v[2])
 end

 stateless(s::SeqModel) = SeqModel(stateless(s.model), s.steps)