projects
/
dagnn.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
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'
require 'nn'
require 'dagnn'
--- torch.setnumthreads(params.nbThreads)
torch.setdefaulttensortype('torch.DoubleTensor')
torch.setdefaulttensortype('torch.DoubleTensor')
-torch.manualSeed(
2
)
+torch.manualSeed(
1
)
function checkGrad(model, criterion, input, target)
local params, gradParams = model:getParameters()
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()
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, 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)
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')
print('Error = ' .. checkGrad(model, nn.MSECriterion(), input, output))
print('Writing /tmp/graph.dot')
-model:
d
ot('/tmp/graph.dot')
+model:
saveD
ot('/tmp/graph.dot')