Torch to ONNX conversion going wrong

deployment
1

Hi Pytorch Team,

I am converting one PyTorch Model(yoloact) trained on my custom dataset to ONNX, the conversion is completing successfully, but in the ONNX model one of the node is not getting connected to the graph.
I have attached the picture of the graph below.

Having said that, the pytorch model Inference is working fine, but not sure why one of the node is not getting connected to the graph in the onnx model.

Can you please assist me to solve this error? Would be a great help from yourside.

I am attaching the screenshot of the graph below:

Screenshot from 2021-05-18 12-17-13
Screenshot from 2021-05-18 12-17-131341×426 28.7 KB

Thanks,
Darshan

2

Looking forward to the Reply.

Regards,
Darshan

3

hey how do you convert Pytorch to Onnx , I’ve tried but no lucks.

4

Hi Darshan could you share your code? It might be possible that you have an unsused constant