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/yolo.h
+++ b/nvdsinfer_custom_impl_Yolo/yolo.h
@@ -74,7 +74,7 @@ public:
    }
    NvDsInferStatus parseModel(nvinfer1::INetworkDefinition& network) override;

-    nvinfer1::ICudaEngine *createEngine (nvinfer1::IBuilder* builder);
+    nvinfer1::ICudaEngine *createEngine (nvinfer1::IBuilder* builder, nvinfer1::IBuilderConfig* config);

 protected:
    const std::string m_NetworkType;