projects
/
dagnn.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
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 = 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(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)
model:setInput(a)
model:setOutput(g)