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?
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?