CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-09-29 08:51:11 +13:00
Code Issues Releases Wiki Activity

Fix error in training script due to missing libcudnn.so

Browse source
This commit is contained in:
nagadomi 2015-12-09 07:04:04 +09:00
parent c4df6f11f4
commit 56582b8401
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
lib/srcnn.lua
View file

@ -9,7 +9,7 @@ function nn.SpatialConvolutionMM:reset(stdv)
self.weight:normal(0, stdv) self.weight:normal(0, stdv)
self.bias:zero() self.bias:zero()
end end
if cudnn then if cudnn and cudnn.SpatialConvolution then
function cudnn.SpatialConvolution:reset(stdv) function cudnn.SpatialConvolution:reset(stdv)
stdv = math.sqrt(2 / ((1.0 + 0.1 * 0.1) * self.kW * self.kH * self.nOutputPlane)) stdv = math.sqrt(2 / ((1.0 + 0.1 * 0.1) * self.kW * self.kH * self.nOutputPlane))
self.weight:normal(0, stdv) self.weight:normal(0, stdv)