diff options
author | Soumith Chintala <soumith@gmail.com> | 2017-02-10 07:39:38 +0300 |
---|---|---|
committer | GitHub <noreply@github.com> | 2017-02-10 07:39:38 +0300 |
commit | c5ab04aaebefa4d0024319dad13a743410485abd (patch) | |
tree | a28feef69b7ba3cf42edf91331054389689b8b61 | |
parent | 99592ae1a5d7afbbd98d75e10aa7f6c3fae806ef (diff) | |
parent | 8b3f833736c81cc2322cdcaa262f2fdde4597c7a (diff) |
Merge pull request #327 from NVIDIA/fix_313
Fixing issue #313
-rw-r--r-- | SpatialFullConvolution.lua | 3 | ||||
-rw-r--r-- | VolumetricFullConvolution.lua | 4 |
2 files changed, 2 insertions, 5 deletions
diff --git a/SpatialFullConvolution.lua b/SpatialFullConvolution.lua index 0ba5cd5..7c9520e 100644 --- a/SpatialFullConvolution.lua +++ b/SpatialFullConvolution.lua @@ -188,8 +188,7 @@ function SpatialFullConvolution:write(f) end function SpatialFullConvolution:clearState() - self:clearDesc() - return nn.Module.clearState(self) + return Convolution.clearState(self) end function SpatialFullConvolution:read(file, version) diff --git a/VolumetricFullConvolution.lua b/VolumetricFullConvolution.lua index d62b37e..a662429 100644 --- a/VolumetricFullConvolution.lua +++ b/VolumetricFullConvolution.lua @@ -206,7 +206,5 @@ function VolumetricFullConvolution:write(f) end function VolumetricFullConvolution:clearState() - self:clearDesc() - nn.utils.clear(self, 'extraBuffer', '_input', '_gradOutput') - return nn.Module.clearState(self) + return Convolution.clearState(self) end |