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

fix exec forward pass

Browse Source
This commit is contained in:
Mike J Innes 2017-03-30 19:25:54 +01:00
parent 4df97bf607
commit 4de61fc377
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

@ -69,8 +69,8 @@ function executor(graph::Graph, input...)
return exec
end
function (exec::Exec)(input)
copy!(exec.args[exec.graph.input[1]], input)
function (exec::Exec)(input...)
foreach(kv -> copy!(exec.args[kv[1]], kv[2]), dictt(exec.graph.input, input))
mx.forward(exec.exec, is_train = true)
mxungroup(exec.graph.output, copy(exec.outs))
end