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 /SpatialCrossMapLRN.lua | |
parent | 501598c58c1af35886a98c0465d96026352c3058 (diff) |
cudnn<>nn conversion
Diffstat (limited to 'SpatialCrossMapLRN.lua')
-rw-r--r-- | SpatialCrossMapLRN.lua | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/SpatialCrossMapLRN.lua b/SpatialCrossMapLRN.lua index 9e92a57..c79f246 100644 --- a/SpatialCrossMapLRN.lua +++ b/SpatialCrossMapLRN.lua @@ -90,9 +90,13 @@ function LRN:updateGradInput(input, gradOutput) return self.gradInput end -function LRN:write(f) +function LRN:clearDesc() self.LRNDesc = nil self.iDesc = nil +end + +function LRN:write(f) + self:clearDesc() local var = {} for k,v in pairs(self) do var[k] = v |