waleedka / hiddenlayer

Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.
MIT License
1.79k stars 266 forks source link

'NoneType' object has no attribute 'nodes' #74

Open samra-irshad opened 4 years ago

samra-irshad commented 4 years ago

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 commented 4 years ago

I solved this problem upgrading from PyTorch 10.1 to 10.2

likyoo commented 4 years ago

a same problem

sumai-1998 commented 2 years ago

1.10.2?