Describe the bug
I try to run tensorboardX/examples/demo_graph.py for jupyter notebook (launched by anaconda navigator) and I get the error seen at Additional context.
I just copy paste the code to notebook from Github.
Minimal runnable code to reproduce the behavior
class SimpleModel(nn.Module):
def init(self):
super(SimpleModel, self).init()
def forward(self, x):
return x * 2
model = SimpleModel()
dummy_input = (torch.zeros(1, 2, 3),)
with SummaryWriter(comment='constantModel') as w:
w.add_graph(model, dummy_input, True)
Describe the bug I try to run tensorboardX/examples/demo_graph.py for jupyter notebook (launched by anaconda navigator) and I get the error seen at Additional context.
I just copy paste the code to notebook from Github.
Minimal runnable code to reproduce the behavior class SimpleModel(nn.Module): def init(self): super(SimpleModel, self).init()
model = SimpleModel() dummy_input = (torch.zeros(1, 2, 3),)
with SummaryWriter(comment='constantModel') as w: w.add_graph(model, dummy_input, True)
Environment Windows, command : conda list | findstr proto, tensor, torch torch -> 1.1.0 libprotobuf -> 3.7.1 protobuf -> 3.5.2 tensorboard -> 1.13.1 tensorboardX -> 1.4 tensorflow -> 1.13.1 tensorflow-base -> 1.13.1 tensorflow-estimator -> 1.13.0 tensorflow-gpu -> 1.13.1
Additional context
AssertionError Traceback (most recent call last)