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")