From bc61702853adbab352660d27b0c37ec3955a6bb9 Mon Sep 17 00:00:00 2001
From: Mike J Innes <mike.j.innes@gmail.com>
Date: Thu, 25 Aug 2016 17:25:53 +0100
Subject: [PATCH] mxmodel tweaks

---
 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 cd1af81c..07b66057 100644
--- a/src/backend/mxnet/model.jl
+++ b/src/backend/mxnet/model.jl
@@ -1,4 +1,4 @@
-type MXModel
+type MXModel <: Model
   model::Any
   params::Dict{Symbol,Any}
   grads::Dict{Symbol,Any}
@@ -59,7 +59,7 @@ end
 
 function (model::MXModel)(input)
   tond!(model.exec.arg_dict[:input], input)
-  mx.forward(model.exec)
+  mx.forward(model.exec, is_train = true)
   fromnd(model.exec.outputs[1])
 end