Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

demo_graph.py error on pytorch 0.4.0

See original GitHub issue

I tried running the demo_graph.py and got the following error.

python demo_graph.py 
Traceback (most recent call last):
  File "demo_graph.py", line 56, in <module>
    w.add_graph(model, (dummy_input, ))
  File "/home/dana/Desktop/tensorboard-pytorch/tensorboardX/writer.py", line 400, in add_graph
    self.file_writer.add_graph(graph(model, input_to_model, verbose))
  File "/home/dana/Desktop/tensorboard-pytorch/tensorboardX/graph.py", line 52, in graph
    trace, _ = torch.jit.trace(model, args)
TypeError: 'function' object is not iterable

The return of torch.jit.trace is this wrapper, from this commit

I’m using the Python 3.6 on Ubuntu 16.04

torch==0.4.0 built from source
tensorboardX==1.1
tensorboard==1.6.0

Edit: Updated my pytorch version

Issue Analytics

State:
Created 6 years ago
Reactions:6
Comments:25 (8 by maintainers)

Top GitHub Comments

9reactions

danakianfarcommented, Mar 17, 2018

Why is this issue closed? Was anyone able to reproduce the issue with pytorch 0.4?

5reactions

WenmuZhoucommented, May 9, 2018

@lanpa I have fixed my error, the code write network to tensorboard must before model.to(device)

model = torchvision.models.AlexNet(num_classes=10)
# 准备写tensorboard, 必须放在'.to(device)'之前，不然会报错
writer = SummaryWriter()
dummy_input = torch.autograd.Variable(torch.rand(1, 3, 227, 227))
writer.add_graph(model=model, input_to_model=(dummy_input, ))

model = model.to(device)