AttributeError: module 'torch.jit' has no attribute '_get_trace_graph'

[AsakusaRinne]

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?

[Rayaction]

same problem with me. did u solve it?
AttributeError: module 'torch.jit' has no attribute '_get_trace_graph'

[bjzhangyong]

Py 3.6, pytorch 1.3.1. And the same problem.
Who can solve it?

[bjzhangyong]

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'