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

new progress integration

Browse Source
This commit is contained in:
Mike J Innes 2016-11-04 22:28:12 +00:00
parent 65531b8689
commit c3d32c395c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/backend/tensorflow/recurrent.jl
View File

@ -55,9 +55,9 @@ function Flux.train!(m::SeqModel, Xs, Ys; epoch = 1, η = 0.1,
Y = placeholder(Float32) Y = placeholder(Float32)
Loss = loss(Y, output)/batchlen/seqlen Loss = loss(Y, output)/batchlen/seqlen
minimize_op = TensorFlow.train.minimize(opt(), Loss) minimize_op = TensorFlow.train.minimize(opt(), Loss)
for e in 1:epoch @progress "training" for e in 1:epoch
info("Epoch $e\n") info("Epoch $e\n")
@progress for (i, (x, y)) in enumerate(zip(Xs,Ys)) @progress "epoch" for (i, (x, y)) in enumerate(zip(Xs,Ys))
out = run(sess, vcat(outstates..., output, Loss, minimize_op), out = run(sess, vcat(outstates..., output, Loss, minimize_op),
merge(Dict(input=>batchone(x), Y=>batchone(y)), merge(Dict(input=>batchone(x), Y=>batchone(y)),
Dict(zip(instates, state)))) Dict(zip(instates, state))))