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

strip more of the trace

Browse Source
This commit is contained in:
Mike J Innes 2017-03-12 18:33:47 +00:00
parent cde74337c8
commit 4b82c57f88
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/mxnet/model.jl
View File

@ -102,8 +102,8 @@ import Base: @get!
executor(m::Model, input) = @get!(m.execs, input, executor(m.graph, input))
function (m::Model)(x)
runrawbatched(x) do x
m.last = exec = @mxerr m.graph.stacks executor(m, size(x))
@mxerr m.graph.stacks runrawbatched(x) do x
m.last = exec = executor(m, size(x))
exec(x)
end
end