Export onnx error

[haikunzhang95]

Dear all,

Hi!

I tried to export the onnx model using torch.onnx.export() but it failed. The error message is as follows：

TyperError: state_dict() takes 1 positional argument but 3 were given

Does anyone know what's going on? Can anyone share the experience of successfully exporting the onnx model?

Thank U all！

Kind Regards

[EricWu-ZL]

Hello, have you solved this problem？