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

[dancingpipi]

https://github.com/ricky40403/PyTransformer/blob/22a0a824be0ef7d4dd65312c4b3e190e4cde4fee/transformers/utils.py#L320

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.