Skip to content

Updated TensorRT Python API in torch2trt#630

Open
andi4191 wants to merge 2 commits intoNVIDIA-AI-IOT:masterfrom
andi4191:anuragd/update_trt_api
Open

Updated TensorRT Python API in torch2trt#630
andi4191 wants to merge 2 commits intoNVIDIA-AI-IOT:masterfrom
andi4191:anuragd/update_trt_api

Conversation

@andi4191
Copy link
Copy Markdown
Contributor

I was seeing issue with TensorRT-7.2.1.6 Python APIs
I fixed it by updating the configuration APIs in torch2trt.py

Signed-off-by: Anurag Dixit <a.dixit91@gmail.com>
Signed-off-by: Anurag Dixit <a.dixit91@gmail.com>
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.

1 participant