CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-05-18 20:02:25 +12:00
Code Issues Releases Wiki Activity

Merge branch 'master' of github.com:nagadomi/waifu2x

Browse source
This commit is contained in:
nagadomi 2015-12-11 16:35:00 +09:00
parent aef9fd8443 56582b8401
commit 7708c7e75a
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.bias:zero()
end
if cudnn then
if cudnn and cudnn.SpatialConvolution then
function cudnn.SpatialConvolution:reset(stdv)
stdv = math.sqrt(2 / ((1.0 + 0.1 * 0.1) * self.kW * self.kH * self.nOutputPlane))
self.weight:normal(0, stdv)