-
Notifications
You must be signed in to change notification settings - Fork 18
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add TensorRT Support to master branch (from #14) #15
Conversation
* [Add]: TensorRT Support * [Fix]: Script improvement * [Fix]: Script improvement * [Fix]: Address the requested changes Co-authored-by: Gpastal24 <gpastal24@users.noreply.github.com>
e3c1ad2
to
2192e85
Compare
Hi @gpastal24 , could you please take a look? |
Hello @stefanopini I tested the colab notebook in the morning , I think it works ok and the detections seemed fine although I still got the warning that trt was linked at a different cudnn. Nonetheless and if someone experiences issues he can always downgrade the trt version. |
Thanks a lot for checking and testing the whole notebook! I'll merge into master |
2192e85
to
6073d82
Compare
Add TensorRT Support from #14 (in branch tensorrt) to master branch.