From 367061fea22736101e08eeb42fe2e28c505b5a65 Mon Sep 17 00:00:00 2001 From: perseusdg Date: Sat, 13 Nov 2021 18:11:46 +0530 Subject: [PATCH] fps fix --- src/NetworkRT.cpp | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/NetworkRT.cpp b/src/NetworkRT.cpp index 743f536..709b9b2 100644 --- a/src/NetworkRT.cpp +++ b/src/NetworkRT.cpp @@ -430,9 +430,13 @@ ILayer* NetworkRT::convert_layer(ITensor *input, Activation *l) { return lRT; } else if(l->act_mode == CUDNN_ACTIVATION_CLIPPED_RELU) { - IPluginV2 *plugin = new ActivationReLUCeiling(l->ceiling); - IPluginV2Layer *lRT = networkRT->addPluginV2(&input, 1, *plugin); + IActivationLayer *lRT = networkRT->addActivation(*input,ActivationType::kCLIP); + //IPluginV2 *plugin = new ActivationReLUCeiling(l->ceiling); + lRT->setAlpha(0); + lRT->setBeta(l->ceiling); checkNULL(lRT); + //IPluginV2Layer *lRT = networkRT->addPluginV2(&input, 1, *plugin); + //checkNULL(lRT); return lRT; } else if(l->act_mode == ACTIVATION_MISH) {