MapTable layer: 'share' parameter is now boolean

1 files changed, 7 insertions, 2 deletions

diff --git a/MapTable.lua b/MapTable.lua
index 90b439c..351eb70 100644
--- a/MapTable.lua
+++ b/MapTable.lua
@@ -2,7 +2,8 @@ local MapTable, parent = torch.class('nn.MapTable', 'nn.Container')
 
 function MapTable:__init(module, shared)
    parent.__init(self)
-   self.shared = shared or {'weight', 'bias', 'gradWeight', 'gradBias'}
+   self.shared = shared or true
+   self.sharedparams = {'weight', 'bias', 'gradWeight', 'gradBias'}
    self.output = {}
    self.gradInput = {}
    self:add(module)
@@ -12,7 +13,11 @@ function MapTable:_extend(n)
    self.modules[1] = self.module
    for i = 2, n do
       if not self.modules[i] then
-         self.modules[i] = self.module:clone(table.unpack(self.shared))
+         if shared then
+           self.modules[i] = self.module:clone(table.unpack(self.sharedparams))
+         else
+           self.modules[i] = self.module:clone(table.unpack(self.sharedparams))
+         end
       end
    end
 end