From 0bb44f5ace657af7f9d885cc1a1924ff6caaa066 Mon Sep 17 00:00:00 2001
From: Mike J Innes <mike.j.innes@gmail.com>
Date: Tue, 21 Feb 2017 13:31:41 +0000
Subject: [PATCH] affine error messages

---
 src/backend/mxnet/graph.jl | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/src/backend/mxnet/graph.jl b/src/backend/mxnet/graph.jl
index aad7aa68..a4f879b9 100644
--- a/src/backend/mxnet/graph.jl
+++ b/src/backend/mxnet/graph.jl
@@ -34,10 +34,11 @@ graph(::typeof(vcat), a...) = graph(cat, 1, a...)
 graph(::Input, x) = x
 
 graph(ctx::Context, d::Affine, x) =
-  mx.FullyConnected(data = x,
-                    num_hidden = size(d.W.x, 2),
-                    weight = var(ctx, d.W),
-                    bias = var(ctx, d.b, size(d.b, 2)))
+  register(ctx,
+    mx.FullyConnected(data = x,
+                      num_hidden = size(d.W.x, 2),
+                      weight = var(ctx, d.W),
+                      bias = var(ctx, d.b, size(d.b, 2))))
 
 # TODO: use actual params}
 graph(ctx::Context, c::Conv2D, x) =
@@ -91,7 +92,7 @@ using Juno
 Juno.errmsg(e::mx.MXError) = e.msg
 
 function errnode(e::mx.MXError)
-  m = match(r"Error in (\w+):", e.msg)
+  m = match(r"Error in (\w+)", e.msg)
   m == nothing && return
   Symbol(m.captures[1])
 end