Update.

[dagnn.git] / test-dagnn.lua

diff --git a/test-dagnn.lua b/test-dagnn.lua
index 366e98f..3801956 100755 (executable)
--- a/test-dagnn.lua
+++ b/test-dagnn.lua
@@ -23,9 +23,8 @@ require 'torch'
 require 'nn'
 require 'dagnn'
 
--- torch.setnumthreads(params.nbThreads)
 torch.setdefaulttensortype('torch.DoubleTensor')
-torch.manualSeed(2)
+torch.manualSeed(1)
 
 function checkGrad(model, criterion, input, target)
    local params, gradParams = model:getParameters()
@@ -92,10 +91,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 +108,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')