Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/torch/nn.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorSergey Zagoruyko <zagoruyko2@gmail.com>2016-02-04 17:53:04 +0300
committerSergey Zagoruyko <zagoruyko2@gmail.com>2016-02-09 21:08:41 +0300
commitb4ebdf2f95ee9f1429825a0d7b0948721e407d82 (patch)
tree77b6183d0c258d07d5f5e6338c48ccb5fc5a877b /Bilinear.lua
parent4a433463fcd17c4ec3242586da7e9fc859feb3ae (diff)
nn.clearState
Diffstat (limited to 'Bilinear.lua')
-rw-r--r--Bilinear.lua5
1 files changed, 5 insertions, 0 deletions
diff --git a/Bilinear.lua b/Bilinear.lua
index fe007c4..3dc6873 100644
--- a/Bilinear.lua
+++ b/Bilinear.lua
@@ -140,3 +140,8 @@ function Bilinear:__tostring__()
(self.bias == nil and ' without bias' or '')
)
end
+
+function Bilinear:clearState()
+ if self.buff then self.buff:set() end
+ return parent.clearState(self)
+end