Open AsakusaRinne opened 4 years ago
same problem with me. did u solve it? AttributeError: module 'torch.jit' has no attribute '_get_trace_graph'
Py 3.6, pytorch 1.3.1. And the same problem. Who can solve it?
Py 3.6, pytorch 1.3.1. And the same problem. Who can solve it?
I modified the '_get_trace_graph' to 'get_trace_graph'. The error information disppeared. But another problem appeared!
d:\download\hiddenlayer-master\hiddenlayer\pytorch_builder.py in import_graph(hl_graph, model, args, input_names, verbose) 76 77 # Loop through nodes and build HL graph ---> 78 for torch_node in torch_graph.nodes(): 79 # Op 80 op = torch_node.kind()
AttributeError: 'NoneType' object has no attribute 'nodes'
I use pytorch1.2.0 to construct a simple network for MNIST to test this visualization tool. There is no error the first time I run it as: `hl_graph=hl.build_graph(MyConvNet,torch.zeros([1,1,28,28]))
hl_graph.theme=hl.graph.THEMES["blue"].copy() hl_graph.save("data/MyConvNet_h1.png",format="png")` However, since the second time it raise an error as the title shows. I click the 'run' button for many times, and only few try succeed. Besides, I couldn't find the png file saved to the directory. Is there a possible solution about this?