'NoneType' object has no attribute 'nodes'

[samra-irshad]

Hi, I am trying to use this package for plotting my model graph, however, the graph does not seem to be built correctly and gives error 'NoneType' object has no attribute 'nodes'. I believe the error is somewhere in 'torch.jit._get_trace_graph' or 'torch.onnx._optimize_trace', but I am unable to figure out?

[rcassani]

I solved this problem upgrading from PyTorch 10.1 to 10.2

[likyoo]

a same problem

[sumai-1998]

1.10.2？