tensorflow / tensorrt

TensorFlow/TensorRT integration
Apache License 2.0
736 stars 226 forks source link

Multiple threads for tf-trt #249

Open hoangmt opened 3 years ago

hoangmt commented 3 years ago

Assume I have two threads, each run same structure, different weights tf-trt models. Will the time cut by half, compared to running two models in sequence? I am experimenting with efficientdet models but two threads approach does not reduce the time that much. Anyone has ever tried multiple threading with tf-trt? Using tensorrt I see the time is reduced by half in several cases but I was not able to convert efficientdet to tensorrt.