diff options
author | Clement Farabet <clement.farabet@gmail.com> | 2013-06-11 05:45:38 +0400 |
---|---|---|
committer | Clement Farabet <clement.farabet@gmail.com> | 2013-06-11 05:45:38 +0400 |
commit | ab8e9f8949f4efc81924d4510d72b509596fbc7b (patch) | |
tree | 2d90ed89f7ea82fdd7bd4669bf5abafe94845518 | |
parent | be1a224ffdbe4aa338bd0f103e92948334cf5e19 (diff) |
More CUDA testing.
-rw-r--r-- | SpatialConvolutionCUDA.lua | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/SpatialConvolutionCUDA.lua b/SpatialConvolutionCUDA.lua index 30c956f..39a2eb1 100644 --- a/SpatialConvolutionCUDA.lua +++ b/SpatialConvolutionCUDA.lua @@ -46,6 +46,7 @@ function SpatialConvolutionCUDA:updateGradInput(input, gradOutput) end function SpatialConvolutionCUDA:accGradParameters(input, gradOutput, scale) + scale = scale or 1 input.nn.SpatialConvolutionCUDA_accGradParameters(self, input, gradOutput, scale) for i = 1,self.nOutputPlane do self.gradBias:narrow(1,i,1):add(scale * gradOutput[i]:sum() ) |