mxmodel tweaks

This commit is contained in:
Mike J Innes 2016-08-25 17:25:53 +01:00
parent afac5d8bfe
commit 526165c897
1 changed files with 2 additions and 2 deletions

View File

@ -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