Relative Content

Tag Archive for pytorchexporttransformer-modelonnxruntime

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.