diff options
| author | Boris Fomitchev <bfomitchev@nvidia.com> | 2017-02-10 03:34:50 +0300 |
|---|---|---|
| committer | Boris Fomitchev <bfomitchev@nvidia.com> | 2017-02-10 03:34:50 +0300 |
| commit | 8b3f833736c81cc2322cdcaa262f2fdde4597c7a (patch) | |
| tree | a28feef69b7ba3cf42edf91331054389689b8b61 | |
| parent | 99592ae1a5d7afbbd98d75e10aa7f6c3fae806ef (diff) | |
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 |