From f8a3b02c1d9c20a7437f34bbf6d30baafc2cde92 Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Wed, 19 Apr 2017 14:48:10 +0100 Subject: [PATCH] fix tf train --- src/backend/tensorflow/model.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/backend/tensorflow/model.jl b/src/backend/tensorflow/model.jl index 63dbc576..ed869504 100644 --- a/src/backend/tensorflow/model.jl +++ b/src/backend/tensorflow/model.jl @@ -54,7 +54,7 @@ function Flux.train!(m::Model, train, test=[]; epoch = 1, η = 0.1, for e in 1:epoch info("Epoch $e\n") @progress for (x, y) in train - y, cur_loss, _ = run(m.session, vcat(m.output, Loss, minimize_op), + y, cur_loss, _ = run(m.session, [m.output, Loss, minimize_op], Dict(m.inputs[1] => batchone(convertel(Float32, x)), Y => batchone(convertel(Float32, y)))) if i % 5000 == 0