diff options
author | Yori Zwols <yori@google.com> | 2015-03-12 17:23:42 +0300 |
---|---|---|
committer | Yori Zwols <yori@google.com> | 2015-03-12 17:23:42 +0300 |
commit | 0e9f7d4e46b0a15f249b4045f86c9d4fbaf9dbcb (patch) | |
tree | 2a08d61a569177d6f45e79ab52de8ace27507756 /test | |
parent | 53e4bedf82ece70c91dabe9b81a780e288fddecb (diff) |
Fixing problem with debug labels for nodes
Diffstat (limited to 'test')
-rw-r--r-- | test/test_nngraph.lua | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/test/test_nngraph.lua b/test/test_nngraph.lua index 4062c5a..9073036 100644 --- a/test/test_nngraph.lua +++ b/test/test_nngraph.lua @@ -80,6 +80,14 @@ function test.test_twoInputs2() checkGradients(module, input) end +function test.test_splitDebugLabels() + local node = nn.Identity()() + node.data.annotations._debugLabel = "node" + local node1, node2 = node:split(2) + assert(node1.data.annotations._debugLabel == "node-1") + assert(node2.data.annotations._debugLabel == "node-2") +end + function test.test_identity() local in1 = nn.Identity()() local in2 = nn.Identity()() |