Print Autograd Graph

41

That’s exactly what i want. Amazing work!

42

You can use TensorBoard for visualization.
TensorBoard is now fully supported in PyTorch version 1.2.0.
More info:
https://pytorch.org/docs/stable/tensorboard.html

43 
g = make_dot(outputs, net.named_parameters)
44

How can we use this tool with a model that have multiple parts in input?
For example, my model takes two inputs like the following coding shows

outputs = net(inputs, tr_angles)

I have tried graph = hl.build_graph(net, inputs, tr_angles) and graph = hl.build_graph(net, (inputs, tr_angles)) and graph = hl.build_graph(net, [inputs, tr_angles])

45

Did you figure this out? For multiple inputs?