From 2ea99797c63e7843a957f0610a75ce65abb51d9c Mon Sep 17 00:00:00 2001 From: lltcggie Date: Wed, 6 Jul 2016 21:30:02 +0900 Subject: [PATCH] =?UTF-8?q?cuDNN=E3=81=AEdll=E3=81=8C=E3=81=AA=E3=81=84?= =?UTF-8?q?=E7=92=B0=E5=A2=83=E3=81=A7GPU=E3=83=A2=E3=83=BC=E3=83=89?= =?UTF-8?q?=E3=81=A7=E5=AE=9F=E8=A1=8C=E3=81=99=E3=82=8B=E3=81=A8=E5=BC=B7?= =?UTF-8?q?=E5=88=B6=E7=B5=82=E4=BA=86=E3=81=99=E3=82=8B=E3=83=90=E3=82=B0?= =?UTF-8?q?=E3=82=92=E4=BF=AE=E6=AD=A3?= 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 080f581..1e3ebac 100644 --- a/common/cNet.cpp +++ b/common/cNet.cpp @@ -362,6 +362,13 @@ Waifu2x::eWaifu2xError cNet::SetParameter(caffe::NetParameter ¶m, const std: else layer_param->mutable_convolution_param()->set_engine(caffe::ConvolutionParameter_Engine_CAFFE); } + else if (type == "Deconvolution") + { + if (process == "cudnn") + layer_param->mutable_convolution_param()->set_engine(caffe::ConvolutionParameter_Engine_CUDNN); + else + layer_param->mutable_convolution_param()->set_engine(caffe::ConvolutionParameter_Engine_CAFFE); + } else if (type == "ReLU") { if (process == "cudnn")