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

fix for empty dicts

Browse Source
This commit is contained in:
Mike J Innes 2017-02-20 23:45:07 +00:00
parent 54011045e7
commit b49daa367f
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

@ -90,6 +90,6 @@ function mx.FeedForward(model::Flux.Model; input = :data, label = :softmax, cont
model = rewrite_softmax(model, label) model = rewrite_softmax(model, label)
vars, stacks, node = tograph(model, mx.Variable(input)) vars, stacks, node = tograph(model, mx.Variable(input))
ff = mx.FeedForward(node, context = context) ff = mx.FeedForward(node, context = context)
ff.arg_params = mxargs(vars) isempty(vars) || (ff.arg_params = mxargs(vars))
return ff return ff
end end