onnx-tensorrt icon indicating copy to clipboard operation
onnx-tensorrt copied to clipboard

pytorch->onnx->tensorrt

Open njustczr opened this issue 3 years ago • 1 comments

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

njustczr avatar Mar 05 '22 13:03 njustczr

Are you able the share the model?

kevinch-nv avatar Mar 12 '22 00:03 kevinch-nv