From 88b5665691555ff4eafc2f125866f6208e2986ad Mon Sep 17 00:00:00 2001 From: alexdrk Date: Fri, 18 Aug 2017 13:58:42 +0300 Subject: [PATCH] Update VolumetricFullConvolution.lua Add clearState --- VolumetricFullConvolution.lua | 5 +++++ 1 file changed, 5 insertions(+) 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