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'


  
    
    
    Print Autograd Graph 
  
  
    Is there a way to visualize the graph of a model similar to what Tensorflow offers?