Browse Source

Fix error in training script due to missing libcudnn.so

nagadomi 9 years ago
parent
commit
56582b8401
1 changed files with 1 additions and 1 deletions
  1. 1 1
      lib/srcnn.lua

+ 1 - 1
lib/srcnn.lua

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