소스 검색

Fix clearState

nagadomi 9 년 전
부모
커밋
fd06a47d56
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  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