From 8c92b43e9633f385a3b71268b177db700ba44290 Mon Sep 17 00:00:00 2001 From: Sam Gross Date: Tue, 22 Dec 2015 10:30:06 -0800 Subject: Nil out the userdata 'oData' before serialization --- SpatialBatchNormalization.lua | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) (limited to 'SpatialBatchNormalization.lua') diff --git a/SpatialBatchNormalization.lua b/SpatialBatchNormalization.lua index bb3b934..9148873 100644 --- a/SpatialBatchNormalization.lua +++ b/SpatialBatchNormalization.lua @@ -11,10 +11,10 @@ function SpatialBatchNormalization:__init(nFeature, eps, momentum, affine) end function SpatialBatchNormalization:createIODescriptors(input) - assert(input:dim() == 4) + assert(input:dim() == 4) assert(torch.typename(self.weight) == 'torch.CudaTensor' and torch.typename(self.bias) == 'torch.CudaTensor', 'Only CUDA tensors are supported for cudnn.SpatialBatchNormalization!') - if not self.iDesc or not self.oDesc or + if not self.iDesc or not self.oDesc or input:size(1) ~= self.iSize[1] or input:size(2) ~= self.iSize[2] or input:size(3) ~= self.iSize[3] or input:size(4) ~= self.iSize[4] then self.iSize = input:size() @@ -30,7 +30,7 @@ local one = torch.FloatTensor({1}); local zero = torch.FloatTensor({0}); function SpatialBatchNormalization:updateOutput(input) - self:createIODescriptors(input) + self:createIODescriptors(input) if self.train then errcheck('cudnnBatchNormalizationForwardTraining', @@ -65,6 +65,7 @@ end function SpatialBatchNormalization:write(f) self.iDesc = nil + self.oDesc = nil self.sDesc = nil local var = {} for k,v in pairs(self) do -- cgit v1.2.3