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

more interp catches

Browse Source
This commit is contained in:
Mike J Innes 2017-02-20 21:49:47 +00:00
parent 242b4d6e23
commit b92252cd5c
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/backend/mxnet/graph.jl
View File

@ -74,8 +74,10 @@ function graph(ctx::Context, model, args...)
interpret(ctx, g, args...) interpret(ctx, g, args...)
end end
graph′(ctx::Context, args...) = @ithrow ctx graph(ctx, args...)
function tograph(model, args...) function tograph(model, args...)
ctx = Context(mux(iline, ilambda, imap, iargs, ituple, graph), ctx = Context(mux(iline, ilambda, imap, iargs, ituple, graph′),
params = Dict(), stacks = Dict()) params = Dict(), stacks = Dict())
out = @icatch graph(ctx, model, args...) out = @icatch graph(ctx, model, args...)
return ctx[:params], ctx[:stacks], out return ctx[:params], ctx[:stacks], out