-
Notifications
You must be signed in to change notification settings - Fork 110
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
Performance about TensorRT #13
Comments
I ever encountered such problem in Jetson TX2. My solution was updating Tensorflow version to 1.11. |
Actually, I used cuda10.0, cudnn7.4.3, Tensorflow 1.13. Is there anything wrong? |
I'm not sure. Maybe you can try to optimize another model first, e.g., in code 1 in this repo, to make sure all settings work. |
Hi @ardianumam @godman2016 |
I haven't solved this problem. Maybe it can accelerate some models, not all. |
I also tried to perform TRT optimization several days ago to SSD MobileNet1 with 1 class. I got 45 FPS in Jetson TX2 for both before & after TRT optimization. My temporary conclusion is: (i) TRT is less optimized for network like MobileNet, maybe due its separable convolution that already performs very small computation so that there is less space for optimization. (ii) When I use more classes (e.g., 80 classes in COCO), there is more difference after TRT optimization (TRT seems optimizing conv. operation for the output prediction which is proportional to the number of classes). |
I tried to accelerate my TensorFlow code by using TensorRT, but it didn't get any improvement?
in frozen_model, num of all_nodes = 893 in TensorRT_model, num of trt_engine_nodes = 0 in TensorRT_model, num of all_nodes = 831
Is there anything wrong? Thanks!
The text was updated successfully, but these errors were encountered: