X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=dagnn.git;a=blobdiff_plain;f=test-dagnn.lua;h=8f92ccf40dcda419b3cb0e25c3d657ab033c0806;hp=1df04e244430f28084047c667ba7e043e3631e6b;hb=be353fdfc2a57172064a024f8cec6015c9d908e5;hpb=116fbcd681f9e097f7acd89f61a15c6b7bd113ce

diff --git a/test-dagnn.lua b/test-dagnn.lua
index 1df04e2..8f92ccf 100755
--- a/test-dagnn.lua
+++ b/test-dagnn.lua
@@ -103,13 +103,13 @@ g = nn.CAddTable()
 model = nn.DAG()
 
 model:addEdge(a, b)
-model:addEdge(b, c)
+model:addEdge(b, nn.Linear(10, 5), nn.ReLU(), nn.Linear(5, 10), c)
 model:addEdge(b, d)
 model:addEdge(c, e)
 model:addEdge(d, e)
 model:addEdge(d, f)
 model:addEdge(e, g)
-model:addEdge(f, g)
+model:addEdge(f, nn.Mul(-1), g)
 
 model:setInput(a)
 model:setOutput(g)