X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=dagnn.git;a=blobdiff_plain;f=test-dagnn.lua;h=462c287d0bfeb3486a0f79e4e463e48c64c659fc;hp=366e98f620b89b1f793284e3317489909a371ce4;hb=fe54a7c5c8425ee9783d82e16a42924e23add457;hpb=9dad4fa1118632bfa02c01e4d6a8a5a129061a54

diff --git a/test-dagnn.lua b/test-dagnn.lua
index 366e98f..462c287 100755
--- 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')