diff options
Diffstat (limited to 'SpatialConvolutionCUDA.lua')
-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() ) |