From 7d1973c3148af5594e4322c8be06b0816c14ee38 Mon Sep 17 00:00:00 2001
From: lltcggie <lltcggie@gmail.com>
Date: Thu, 25 Oct 2018 20:57:45 +0900
Subject: [PATCH] =?UTF-8?q?CUDA=E3=81=8C=E4=BD=BF=E3=81=88=E3=81=AA?=
 =?UTF-8?q?=E3=81=84=E7=92=B0=E5=A2=83=E3=81=A7CPU=E3=81=A7UpResNet10?=
 =?UTF-8?q?=E3=82=92=E5=8B=95=E4=BD=9C=E3=81=95=E3=81=9B=E3=82=8B=E3=81=A8?=
 =?UTF-8?q?=E3=82=AF=E3=83=A9=E3=83=83=E3=82=B7=E3=83=A5=E3=81=99=E3=82=8B?=
 =?UTF-8?q?=E3=81=AE=E3=82=92=E4=BF=AE=E6=AD=A3=20resolved=20#126?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 common/cNet.cpp | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/common/cNet.cpp b/common/cNet.cpp
index 0e84f3c..902858d 100644
--- a/common/cNet.cpp
+++ b/common/cNet.cpp
@@ -390,6 +390,13 @@ Waifu2x::eWaifu2xError cNet::SetParameter(caffe::NetParameter &param, const std:
 			else
 				layer_param->mutable_relu_param()->set_engine(caffe::ReLUParameter_Engine_CAFFE);
 		}
+		else if (type == "Sigmoid")
+		{
+			if (process == "cudnn")
+				layer_param->mutable_sigmoid_param()->set_engine(caffe::SigmoidParameter_Engine_CUDNN);
+			else
+				layer_param->mutable_sigmoid_param()->set_engine(caffe::SigmoidParameter_Engine_CAFFE);
+		}
 	}
 
 	return Waifu2x::eWaifu2xError_OK;