onnx-tensorrt
onnx-tensorrt copied to clipboard
pytorch->onnx->tensorrt
when i test onnx and tensorrt, the mse of the two models is about 0.6, i can not find the reason, anyone can help me?
Inference time with the TensorRT engine: 0.0007791519165039062 Inference time with the pytorch model: 0.25454282760620117 MSE Error = 0.5628564357757568
Are you able the share the model?