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

fix for latest mxnet api

Browse Source
This commit is contained in:
Mike J Innes 2017-03-08 16:20:33 +00:00
parent 6a3bed1e61
commit 3b004bac7d
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/graph.jl
View File

@ -36,7 +36,7 @@ graph(::Input, x) = x
graph(ctx::Context, d::Affine, x) = graph(ctx::Context, d::Affine, x) =
!ctx[:feedforward] ? invoke(graph, (Context, Any, typeof(x)), ctx, d, x) : !ctx[:feedforward] ? invoke(graph, (Context, Any, typeof(x)), ctx, d, x) :
register(ctx, register(ctx,
mx.FullyConnected(x, mx.FullyConnected(mx.SymbolicNode, data = x,
num_hidden = size(d.W.x, 2), num_hidden = size(d.W.x, 2),
weight = var(ctx, AlterParam(d.W, false, false)), weight = var(ctx, AlterParam(d.W, false, false)),
bias = var(ctx, AlterParam(d.b, true, false)))) bias = var(ctx, AlterParam(d.b, true, false))))