Cosmetics.

[dagnn.git] / test-dagnn.lua

diff --git a/test-dagnn.lua b/test-dagnn.lua
index 366e98f..462c287 100755 (executable)
--- a/test-dagnn.lua
+++ b/test-dagnn.lua
@@ -92,10 +92,9 @@ c = nn.Linear(10, 15)
 d = nn.CMulTable()
 e = nn.CAddTable()
 
-model:connect(a, b)
+model:connect(a, b, c)
 model:connect(b, nn.Linear(10, 15), nn.ReLU(), d)
 model:connect(d, e)
-model:connect(b, c)
 model:connect(c, d)
 model:connect(c, nn.Mul(-1), e)
 
@@ -110,4 +109,4 @@ output:uniform()
 print('Error = ' .. checkGrad(model, nn.MSECriterion(), input, output))
 
 print('Writing /tmp/graph.dot')
-model:dot('/tmp/graph.dot')
+model:saveDot('/tmp/graph.dot')