From 4de61fc377028ae6b6d927bdbde494d3cf0d1a69 Mon Sep 17 00:00:00 2001
From: Mike J Innes <mike.j.innes@gmail.com>
Date: Thu, 30 Mar 2017 19:25:54 +0100
Subject: [PATCH] fix exec forward pass

---
 src/backend/mxnet/model.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/backend/mxnet/model.jl b/src/backend/mxnet/model.jl
index c5c8bb7f..e03d4e0f 100644
--- a/src/backend/mxnet/model.jl
+++ b/src/backend/mxnet/model.jl
@@ -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