mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

fps fix

Browse Source
This commit is contained in:
perseusdg
2021-11-13 18:11:46 +05:30
parent d488c3bd17
commit 367061fea2
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/NetworkRT.cpp
+6 -2
View File
@@ -430,9 +430,13 @@ ILayer* NetworkRT::convert_layer(ITensor *input, Activation *l) {
return lRT; return lRT;
} }
else if(l->act_mode == CUDNN_ACTIVATION_CLIPPED_RELU) { else if(l->act_mode == CUDNN_ACTIVATION_CLIPPED_RELU) {
IPluginV2 *plugin = new ActivationReLUCeiling(l->ceiling); IActivationLayer *lRT = networkRT->addActivation(*input,ActivationType::kCLIP);
IPluginV2Layer *lRT = networkRT->addPluginV2(&input, 1, *plugin); //IPluginV2 *plugin = new ActivationReLUCeiling(l->ceiling);
lRT->setAlpha(0);
lRT->setBeta(l->ceiling);
checkNULL(lRT); checkNULL(lRT);
//IPluginV2Layer *lRT = networkRT->addPluginV2(&input, 1, *plugin);
//checkNULL(lRT);
return lRT; return lRT;
} }
else if(l->act_mode == ACTIVATION_MISH) { else if(l->act_mode == ACTIVATION_MISH) {