Fix network-mode

Fix FP32, FP16 and INT8 modes
2022-02-08 22:14:20 -03:00
parent 0fdaa1142a
commit d9c3cbf41d
3 changed files with 3 additions and 6 deletions
--- a/nvdsinfer_custom_impl_Yolo/nvdsinfer_yolo_engine.cpp
+++ b/nvdsinfer_custom_impl_Yolo/nvdsinfer_yolo_engine.cpp
@@ -107,7 +107,7 @@ bool NvDsInferYoloCudaEngineGet(nvinfer1::IBuilder * const builder,
    }

    Yolo yolo(networkInfo);
-    cudaEngine = yolo.createEngine (builder);
+    cudaEngine = yolo.createEngine (builder, builderConfig);
    if (cudaEngine == nullptr)
    {
        std::cerr << "Failed to build CUDA engine on "