Added DAG:dot() to generate a dot file for visualization.

[dagnn.git] / test-dagnn.lua

diff --git a/test-dagnn.lua b/test-dagnn.lua
index 3dea310..53302fd 100755 (executable)
--- a/test-dagnn.lua
+++ b/test-dagnn.lua
@@ -108,3 +108,5 @@ local output = model:updateOutput(input):clone()
 output:uniform()
 
 print('Error = ' .. checkGrad(model, nn.MSECriterion(), input, output))
+
+model:dot('/tmp/graph.dot')