nagadomi 9 lat temu
rodzic
commit
fd06a47d56
1 zmienionych plików z 2 dodań i 2 usunięć
  1. 2 2
      lib/srcnn.lua

+ 2 - 2
lib/srcnn.lua

@@ -19,10 +19,10 @@ end
 
 function nn.SpatialConvolutionMM:clearState()
    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
    if self.gradBias then
-      self.gradBias = torch.Tensor(self.nOutputPlane):typeAs(self.gradBias):zero()
+      self.gradBias:resize(self.nOutputPlane):zero()
    end
    return nn.utils.clear(self, 'finput', 'fgradInput', '_input', '_gradOutput', 'output', 'gradInput')
 end