Print torch graph

1

I tried the code posted here. But I got the following error. Does anybody have any code that is working for the latest pytorch? Thanks.

/tmp$ python viz_net_pytorch.py 
Variable containing:
-1.5643e-01  2.2547e-01 -2.9428e-01  ...  -8.9741e-02  3.4645e-01  1.0025e-02
[torch.FloatTensor of size 1x1000]

Traceback (most recent call last):
  File "viz_net_pytorch.py", line 42, in <module>
    g = make_dot(y)
  File "viz_net_pytorch.py", line 33, in make_dot
    add_nodes(var.creator)
  File "/usr/local/lib/python2.7/site-packages/torch/autograd/variable.py", line 67, in __getattr__
    return object.__getattribute__(self, name)
AttributeError: 'Variable' object has no attribute 'creator'
2

.creator was changed to .grad_fn a while ago.
You could try this script to visualize the graph.