Add models benchmarks

Add models benchmarks Minor fixes
2022-02-10 00:52:40 -03:00
parent 5b9b1f77c8
commit ff785458dd
5 changed files with 68 additions and 26 deletions
--- a/nvdsinfer_custom_impl_Yolo/layers/convolutional_layer.cpp
+++ b/nvdsinfer_custom_impl_Yolo/layers/convolutional_layer.cpp
@@ -14,6 +14,7 @@ nvinfer1::ILayer* convolutionalLayer(
    int& weightPtr,
    std::string weightsType,
    int& inputChannels,
+    float eps,
    nvinfer1::ITensor* input,
    nvinfer1::INetworkDefinition* network)
 {
@@ -156,7 +157,7 @@ nvinfer1::ILayer* convolutionalLayer(
            }
            for (int i = 0; i < filters; ++i)
            {
-                bnRunningVar.push_back(sqrt(weights[weightPtr] + 1.0e-3));
+                bnRunningVar.push_back(sqrt(weights[weightPtr] + eps));
                weightPtr++;
            }
            trtWeights.push_back(convWt);