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/activation_layer.cpp
+++ b/nvdsinfer_custom_impl_Yolo/layers/activation_layer.cpp
@@ -81,7 +81,7 @@ nvinfer1::ILayer* activationLayer(
        mish->setName(mishLayerName.c_str());
        output = mish;
    }
-    else if (activation == "silu")
+    else if (activation == "silu" || activation == "swish")
    {
        nvinfer1::IActivationLayer* sigmoid = network->addActivation(
            *input, nvinfer1::ActivationType::kSIGMOID);