From a99bb03830c22556da9844a0affcd96dce2d756f Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Sun, 30 Oct 2016 10:55:07 +0000 Subject: [PATCH] gradients are slow --- src/backend/tensorflow/recurrent.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/backend/tensorflow/recurrent.jl b/src/backend/tensorflow/recurrent.jl index bc75e760..23afa265 100644 --- 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