From b7af24980e7b8aa5ad015d1db679628b8d9bb0c5 Mon Sep 17 00:00:00 2001
From: thnkinbtfly <tbvj5914@gmail.com>
Date: Mon, 14 Dec 2020 05:20:58 +0000
Subject: [PATCH] support sigmoid (logistic) activation

---
 src/DarknetParser.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/DarknetParser.cpp b/src/DarknetParser.cpp
index 7b5410c..d8c186a 100644
--- a/src/DarknetParser.cpp
+++ b/src/DarknetParser.cpp
@@ -185,6 +185,7 @@ namespace tk { namespace dnn {
         if(netLayers.size() > 0 && f.activation != "linear") {
             tkdnnActivationMode_t act;
             if(f.activation == "relu") act = tkdnnActivationMode_t(CUDNN_ACTIVATION_RELU);
+            else if(f.activation == "logistic") act = tkdnnActivationMode_t(CUDNN_ACTIVATION_SIGMOID);
             else if(f.activation == "leaky") act = tk::dnn::ACTIVATION_LEAKY;
             else if(f.activation == "mish") act = tk::dnn::ACTIVATION_MISH;
             else { FatalError("activation not supported: " + f.activation); }