# Contact <francois.fleuret@idiap.ch> for comments & bug reports #
#########################################################################
+import subprocess
+
from torch import nn
from torch.nn import functional as fn
from torch import Tensor
loss = criterion(output, target)
agtree2dot.save_dot(loss,
- { input: 'input', target: 'target', loss: 'loss' },
+ {
+ input: 'input',
+ target: 'target',
+ loss: 'loss',
+ mlp.fc1.weight: 'weight1',
+ mlp.fc1.bias: 'bias1',
+ mlp.fc2.weight: 'weight2',
+ mlp.fc2.bias: 'bias2',
+ },
open('./mlp.dot', 'w'))
-print('Generated mlp.dot. You can convert it to pdf with')
-print('> dot mlp.dot -Lg -T pdf -o mlp.pdf')
+print('Generated mlp.dot')
+
+try:
+ subprocess.check_call(["dot", "mlp.dot", "-Lg", "-T", "pdf", "-o", "mlp.pdf" ])
+except subprocess.CalledProcessError:
+ print('Calling the dot command failed. Is Graphviz installed?')
+ sys.exit(1)
+
+print('Generated mlp.pdf')