How do you generate this form of model

[ruotianluo]

I'm wondering where i can find the standard code to remove the gradBias and gradWeight and generate a unpack function to recover the model.

[szagoruyko]

for k,v in ipairs(net:listModules()) do
  v.gradWeight = nil
  v.gradBias = nil
end

local ts = {
  model = net:float(),
  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
}

torch.save('model.t7', ts)

[ruotianluo]

Thank you