diff options
| author | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-01-15 16:25:09 +0300 |
|---|---|---|
| committer | soumith <soumith@gmail.com> | 2016-01-26 19:02:48 +0300 |
| commit | d96d54fbf779648f0ce156ecb5d8ae3108a7b0e6 (patch) | |
| tree | eeb9097458cd024336307fa65013e3927e9b4dab /SpatialConvolution.lua | |
| parent | 501598c58c1af35886a98c0465d96026352c3058 (diff) | |
cudnn<>nn conversion
Diffstat (limited to 'SpatialConvolution.lua')
| -rw-r--r-- | SpatialConvolution.lua | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/SpatialConvolution.lua b/SpatialConvolution.lua index e13b1be..88dee31 100644 --- a/SpatialConvolution.lua +++ b/SpatialConvolution.lua @@ -452,7 +452,7 @@ function SpatialConvolution:accGradParameters(input, gradOutput, scale) end end -function SpatialConvolution:write(f) +function SpatialConvolution:clearDesc() self.weightDesc = nil self.biasDesc = nil self.convDesc = nil @@ -465,6 +465,11 @@ function SpatialConvolution:write(f) self.bwdFilterAlgType = nil self.extraBuffer = nil self.extraBufferSizeInBytes = nil + self.scaleT = nil +end + +function SpatialConvolution:write(f) + self:clearDesc() local var = {} for k,v in pairs(self) do var[k] = v |