diff --git a/VolumetricFullConvolution.lua b/VolumetricFullConvolution.lua index 60843e73d..d1d51e98b 100644 --- a/VolumetricFullConvolution.lua +++ b/VolumetricFullConvolution.lua @@ -223,3 +223,8 @@ function VolumetricFullConvolution:__tostring__() end return s .. ')' end + +function VolumetricFullConvolution:clearState() + nn.utils.clear(self, 'finput', 'fgradInput', '_input', '_gradOutput') + return parent.clearState(self) +end