TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feedforward fix

Browse Source
This commit is contained in:
Mike J Innes 2017-06-09 18:54:35 +01:00
parent 9b18fd639a
commit 9c8dbb6b4b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/backend/mxnet/model.jl
View File

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