Skip to content

Conversation

stevenlix
Copy link
Contributor

  1. Update Onnx-TensorRT parser to include latest fix
    The fix solved the issue of accuracy loss in some opset11 models, such as yolov4 in onnx model zoo.
    The reason is certain new attributes of opset11 Resize operator are not supported in TensorRT7.1 and also are not handled properly in the parser.
  2. Move cudaDeviceSynchronize right after enquenV2 to make sure TRT outputs are synced before fetching

@stevenlix stevenlix merged commit 186f066 into master Oct 10, 2020
@stevenlix stevenlix deleted the stevenlix/accuracy branch October 10, 2020 04:49
tianleiwu pushed a commit that referenced this pull request Oct 10, 2020
tianleiwu pushed a commit that referenced this pull request Oct 12, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants