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

print last line for TF errors

Browse Source
This commit is contained in:
Mike J Innes 2016-12-20 18:15:42 +00:00
parent b580c2e4a7
commit 9775baddd3
1 changed files with 19 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/backend/tensorflow/model.jl
View File

@ -36,10 +36,26 @@ function batch(xs)
Batch{T,B}(xs) Batch{T,B}(xs)
end end
function runmodel(m, args...) function tferr(model::Model, e)
m = match(r"Node: ([\w\d]+) =", string(e.status))
m == nothing && return
node = m.captures[1]
if haskey(model.stacks, node)
l = model.stacks[node][end]
println("TensorFlow error occured at $(l.file):$(l.line)")
end
end
function runmodel(m::Model, args...)
@assert length(args) == length(m.inputs) @assert length(args) == length(m.inputs)
output = run(m.session, m.output, Dict(zip(m.inputs, args))) try
ismultioutput(m) ? (batch.(output)...,) : batch(output) output = run(m.session, m.output, Dict(zip(m.inputs, args)))
ismultioutput(m) ? (batch.(output)...,) : batch(output)
catch e
isa(e, TensorFlow.TFException) || rethrow(e)
tferr(m, e)
rethrow(e)
end
end end
function (m::Model)(args::Batch...) function (m::Model)(args::Batch...)