Mirror/waifu2x-caffe
1
0
Fork 0
mirror of https://github.com/lltcggie/waifu2x-caffe.git synced 2025-06-26 05:32:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

CUDAが使えない環境でCPUでUpResNet10を動作させるとクラッシュするのを修正 resolved #126

Browse Source
This commit is contained in:
lltcggie 2018-10-25 20:57:45 +09:00
parent de6de69ac6
commit 7d1973c314
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
common/cNet.cpp
View File

@ -390,6 +390,13 @@ Waifu2x::eWaifu2xError cNet::SetParameter(caffe::NetParameter &param, const std:
else else
layer_param->mutable_relu_param()->set_engine(caffe::ReLUParameter_Engine_CAFFE); 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; return Waifu2x::eWaifu2xError_OK;