浏览代码

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