Add an optional `engineDir` parameter to the `buildLoadNetwork` function

cyrusbehr / tensorrt-cpp-api

TensorRT C++ API Tutorial

MIT License

543 stars 66 forks source link

Add an optional `engineDir` parameter to the `buildLoadNetwork` function #62

Closed thomaskleiven closed 1 month ago

thomaskleiven commented 1 month ago

Closes #61

Description: This PR adds an optional engineDir parameter to the buildLoadNetwork function. This parameter allows users to specify the directory where .engine models are stored.

Changes:

Added engineDir parameter to buildLoadNetwork.
Updated the function to store/load the engine file from the specified directory.

cyrusbehr commented 1 month ago

Decided to implement this as Option.engineFileDir. See this commit where it was added. I'm still going to give you credit in the project readme though since you put in the work to get this PR up.

cyrusbehr commented 1 month ago

@all-contributors please add @thomaskleiven for code

allcontributors[bot] commented 1 month ago

@cyrusbehr

I've put up a pull request to add @thomaskleiven! :tada: