diff options
author | Clement Farabet <clement.farabet@gmail.com> | 2011-07-13 20:35:35 +0400 |
---|---|---|
committer | Clement Farabet <clement.farabet@gmail.com> | 2011-07-13 20:35:35 +0400 |
commit | 6ee12a28e5d7417d1d5908e4cbd20707bddb6711 (patch) | |
tree | d8aa1dae26ac85385e747ebb715f3f76ab75a389 /init.lua | |
parent | ceb6c62bb0ac1a3f3c6e01870dc4b71ea7a232e3 (diff) |
Added convenient empty function()
Diffstat (limited to 'init.lua')
-rw-r--r-- | init.lua | 27 |
1 files changed, 27 insertions, 0 deletions
@@ -98,3 +98,30 @@ torch.include('nnx', 'StochasticTrainer.lua') torch.include('nnx', 'DataSet.lua') torch.include('nnx', 'DataList.lua') torch.include('nnx', 'DataSetLabelMe.lua') + +-- helpers: +function nnx.empty(module) + if module.modules then + -- find submodules in classic containers 'modules' + for _,module in ipairs(module.modules) do + nnx.empty(module) + end + else + -- find arbitrary submodules + for k,entry in pairs(module) do + local type = torch.typename(entry) + if type and type:find('^nn.') then + nnx.empty(entry) + end + end + end + -- empty module + if module.output and module.output.resize then + module.output:resize() + module.output:storage():resize(0) + end + if module.gradInput and module.gradInput.resize then + module.gradInput:resize() + module.gradInput:storage():resize(0) + end +end |