diff options
author | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2015-10-10 05:08:51 +0300 |
---|---|---|
committer | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2015-10-10 05:08:51 +0300 |
commit | 6b4b141f8e20b85563d4a25316329fdb33b9b138 (patch) | |
tree | bcba156b109535809804bcdfe9fd0e8507157236 | |
parent | c583b7502316a62705b6a355afcd4dcf556f96f2 (diff) |
fix serialization of SpatialConvolution
-rw-r--r-- | SpatialConvolution.lua | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/SpatialConvolution.lua b/SpatialConvolution.lua index c67130a..3ebacc4 100644 --- a/SpatialConvolution.lua +++ b/SpatialConvolution.lua @@ -24,6 +24,8 @@ function SpatialConvolution:__init(nInputPlane, nOutputPlane, self.weight = torch.Tensor(nOutputPlane, nInputPlane/self.groups, kW, kH) self.gradWeight = torch.Tensor(nOutputPlane, nInputPlane/self.groups, kW, kH) self:reset() + -- should nil for serialization, the reset will still work + self.reset = nil self.iSize = torch.LongStorage(4):fill(0) end |