DAG:addEdge now allows to add a bunch of edges in one shot. This allows to use anonym...

[dagnn.git] / test-dagnn.lua

diff --git a/test-dagnn.lua b/test-dagnn.lua
index 1df04e2..8f92ccf 100755 (executable)
--- 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)