Merge branch 'master' of github.com:clementfarabet/lua---nnx

1 files changed, 8 insertions, 3 deletions

diff --git a/SoftMaxTree.lua b/SoftMaxTree.lua
index af4be64..6633359 100644
--- a/SoftMaxTree.lua
+++ b/SoftMaxTree.lua
@@ -264,7 +264,11 @@ function SoftMaxTree:zeroGradParameters()
    for k,gradParam in pairs(gradParams) do
       gradParam:zero()
    end
-   self.updates = {}
+   -- loop is used instead of 'self.updates = {}'
+   -- to handle the case when updates are shared
+   for k,v in pairs(self.updates) do
+      self.updates[k] = nil
+   end
 end
 
 function SoftMaxTree:type(type)
@@ -325,9 +329,10 @@ function SoftMaxTree:sharedClone()
    smt.childParent = self.childParent
    smt.maxFamilyPath = self.maxFamilyPath
    smt.maxDept = self.maxDept
+   smt.updates = self.updates
    if not self.accUpdate then
-      smt.gradWeight = self.gradWeight:clone()
-      smt.gradBias = self.gradBias:clone()
+      smt.gradWeight = self.gradWeight
+      smt.gradBias = self.gradBias
    end
    if type == 'torch.CudaTensor' then
       smt.parentChildrenCuda = self.parentChildrenCuda