Bläddra i källkod

Fix clearState

nagadomi 9 år sedan
förälder
incheckning
fd06a47d56
1 ändrade filer med 2 tillägg och 2 borttagningar
  1. 2 2
      lib/srcnn.lua

+ 2 - 2
lib/srcnn.lua

@@ -19,10 +19,10 @@ end
 
 
 function nn.SpatialConvolutionMM:clearState()
 function nn.SpatialConvolutionMM:clearState()
    if self.gradWeight then
    if self.gradWeight then
-      self.gradWeight = torch.Tensor(self.nOutputPlane, self.nInputPlane * self.kH * self.kW):typeAs(self.gradWeight):zero()
+      self.gradWeight:resize(self.nOutputPlane, self.nInputPlane * self.kH * self.kW):zero()
    end
    end
    if self.gradBias then
    if self.gradBias then
-      self.gradBias = torch.Tensor(self.nOutputPlane):typeAs(self.gradBias):zero()
+      self.gradBias:resize(self.nOutputPlane):zero()
    end
    end
    return nn.utils.clear(self, 'finput', 'fgradInput', '_input', '_gradOutput', 'output', 'gradInput')
    return nn.utils.clear(self, 'finput', 'fgradInput', '_input', '_gradOutput', 'output', 'gradInput')
 end
 end