search

cyrusbehr / tensorrt-cpp-api

TensorRT C++ API Tutorial
MIT License
543 stars 66 forks source link

Feature Request: Add enginePath Argument to Engine::build Method #61

Closed thomaskleiven closed 1 month ago

thomaskleiven commented 1 month ago

First, thank you for maintaining this project—it's a really great library!

Would it be a bad idea to add an optional enginePath argument to the Engine::build method? This would allow users to specify where to store .engine models, defaulting to a reasonable path for backwards compatibility.

This feature is particularly useful for Dockerized applications, enabling better file organization and management.

cyrusbehr commented 1 month ago

I like the idea. Are you willing to open a PR with the change please?

thomaskleiven commented 1 month ago

Sure, happy to open a PR within the next few days.

cyrusbehr commented 1 month ago

Thanks for your work with this.