gradients are slow

2016-10-30 10:55:07 +00:00 · 2016-10-30 10:55:07 +00:00 · a99bb03830
commit a99bb03830
parent ec1950b466
1 changed files with 1 additions and 1 deletions
--- a/src/backend/tensorflow/recurrent.jl
+++ b/src/backend/tensorflow/recurrent.jl
@ -18,7 +18,7 @@ function tf(model::Flux.Unrolled)
  SeqModel(
    Model(model, sess, params,
          [instates..., input], [outstates..., output],
-          [gradients(output, input)]),
+          [placeholder(Float32)]),
    batchone.(model.state))
 end