Hello,
So I want to create a graph for my Unet model and with pytorch1.1.0 and tensorboardx, it shows me
‘’‘pytorch version is too old, how about build by yourself?’’’ this remark and moves forward.
I have created model on tensorboardx with pytorch1.0.6 post2.
I tried to use torch.utils.tensorboard , but it gives me error too.
Here my summary writer code.
writer1 = SummaryWriter()
dummy_inp = torch.randn(1, 3, 128, 128)
model_test.to('cpu')
writer1.add_graph(model_test, model_test(torch.randn(3, 3, 128, 128, requires_grad=True)))
model_test.to(device)
If you have a working example of creating models with pytorch1.1.0 for skip connections one please refer me to that too.
Thank you.