fix for empty dicts

2017-02-20 23:45:07 +00:00 · 2017-02-20 23:45:07 +00:00 · b49daa367f
commit b49daa367f
parent 54011045e7
1 changed files with 1 additions and 1 deletions
--- a/src/backend/mxnet/model.jl
+++ b/src/backend/mxnet/model.jl
@ -90,6 +90,6 @@ function mx.FeedForward(model::Flux.Model; input = :data, label = :softmax, cont
  model = rewrite_softmax(model, label)
  vars, stacks, node = tograph(model, mx.Variable(input))
  ff = mx.FeedForward(node, context = context)
-  ff.arg_params = mxargs(vars)
+  isempty(vars) || (ff.arg_params = mxargs(vars))
  return ff
 end