Hello, I encountered some issues while exporting MSA to onnx.
I used MSA to train the REFUGE dataset and set multiask_output to 2 as required, successfully saving checkpoint best. pth.
But when I wanted to export checkpoint_best.pth as onnx, it prompted me that the forward parameter in torch's onnx was missing the multiask_output parameter. I would like to inquire about how to solve this problem.
puallee
commented
7 months ago
Hello, I encountered some issues while exporting MSA to onnx.
I used MSA to train the REFUGE dataset and set multiask_output to 2 as required, successfully saving checkpoint best. pth.
But when I wanted to export checkpoint_best.pth as onnx, it prompted me that the forward parameter in torch's onnx was missing the multiask_output parameter. I would like to inquire about how to solve this problem.