unpack.lua

ts = {
  model = net,
  unpack = function(self)
    for k,v in ipairs(self.model:listModules()) do
      if v.weight and not v.gradWeight then
        v.gradWeight = v.weight:clone()
        v.gradBias = v.bias:clone()
      end
    end
    return self.model
  end
}