diff --git a/src/backend/mxnet/model.jl b/src/backend/mxnet/model.jl index 24ce7cfe..d27a94b6 100644 --- a/src/backend/mxnet/model.jl +++ b/src/backend/mxnet/model.jl @@ -10,8 +10,6 @@ end tond(xs::AArray) = copy!(mx.zeros(size(xs)), xs) -fromnd(xs::mx.NDArray) = copy(xs) - ndzero!(xs::mx.NDArray) = copy!(xs, mx.zeros(size(xs))) function mxargs(args) @@ -50,13 +48,13 @@ end function (model::MXModel)(input) copy!(model.exec.arg_dict[:input], input) mx.forward(model.exec, is_train = true) - fromnd(model.exec.outputs[1]) + copy(model.exec.outputs[1]) end function Flux.back!(model::MXModel, Δ, x) ndzero!(model.grads[:input]) mx.backward(model.exec, tond(Δ)) - fromnd(model.grads[:input]) + copy(model.grads[:input]) end function Flux.update!(model::MXModel, η)