search

szagoruyko / pytorchviz

A small package to create visualizations of PyTorch execution graphs
MIT License
3.24k stars 279 forks source link

function object is not iterable error in master #6

Closed szagoruyko closed 6 years ago

szagoruyko commented 6 years ago

Running master 0.4.0a0+2df578a, get this while running notebook:

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-4-fe552e8dc282> in <module>()
----> 1 trace, _ = torch.jit.trace(model, args=(x,))
      2 make_dot_from_trace(trace)

TypeError: 'function' object is not iterable
kenfehling commented 6 years ago

Is that related to this issue?

https://github.com/lanpa/tensorboard-pytorch/pull/106

szagoruyko commented 6 years ago

fixed now