From 9c8dbb6b4b7bb162057f82a64f90eca5c81ffec3 Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Fri, 9 Jun 2017 18:54:35 +0100 Subject: [PATCH] feedforward fix --- src/backend/mxnet/model.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/backend/mxnet/model.jl b/src/backend/mxnet/model.jl index 6a09d44f..2488e3c1 100644 --- a/src/backend/mxnet/model.jl +++ b/src/backend/mxnet/model.jl @@ -145,7 +145,7 @@ end function FeedForward(model; input = :data, label = :softmax, ctx = mx.cpu()) model = rewrite_softmax(model, label) graph = tograph(model, input, feedforward=true) - ff = mx.FeedForward(graph.output, context = context) + ff = mx.FeedForward(graph.output, context = ctx) isempty(graph.params) || (ff.arg_params = ndparams(mxparams(graph.params, ctx))) return ff end