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

gradients are slow

Browse Source
This commit is contained in:
Mike J Innes 2016-10-30 10:55:07 +00:00
parent ec1950b466
commit a99bb03830
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/backend/tensorflow/recurrent.jl
View File

@ -18,7 +18,7 @@ function tf(model::Flux.Unrolled)
SeqModel( SeqModel(
Model(model, sess, params, Model(model, sess, params,
[instates..., input], [outstates..., output], [instates..., input], [outstates..., output],
[gradients(output, input)]), [placeholder(Float32)]),
batchone.(model.state)) batchone.(model.state))
end end