_build_graph error: while model contain module that have more then one input

[dancingpipi]

PyTransformer/transformers/utils.py

Line 320 in 22a0a82

# general layer should has only one input?

how can I solve the problem?

[ricky40403]

Hi, z13974509906:

The purpose we build this repository is for general torch.nn layers. So it rarely required multiple inputs.

Could you explain the detail or provide some examples for testing the error.

Regards,
Ricky

[liming312]

Hi @ricky40403 . For example, this one.

[ricky40403]

Hmm, I did not consider the non-tensor output.

If you are in a hurry to use the visualization/summary part, you may try some rules to change the rule of the graph.
Or, there is one way to parse the model by using onnx and convert as the graph and visualize them.