Onnxruntime Test Error after Successfully Converting Model to ONNX
I have a simple PyTorch model that I’m attempting to convert to ONNX format. The forward()
function consists of a call to nn.transformer.encoder()
. The conversion to ONNX using torch.onnx.export()
completes successfully. However, when I test the model using onnxruntime_test
, it fails, except for specific input cases.