TensorRT
TensorRT copied to clipboard
How to get the profile by python?
Description
I can get the profile by trtexce:
./trtexec --loadEngine=debug_fp16.trt --dumpProfile --shapes=input:1x3x512x512 --exportProfile=debug_profile
How can I get the debug_profile by python when I convert onnx to trt engine by python?
I haven't used it before but I guess https://docs.nvidia.com/deeplearning/tensorrt/api/python_api/infer/Core/ExecutionContext.html#tensorrt.IExecutionContext.report_to_profiler is the answer, @nvpohanh may know more about it.
closing since no activity for more than 3 weeks, please reopen if you still have question, thanks!