nn.MM
需要一个将乘以的矩阵的表参数。在我的例子中,其中一个矩阵是一些先前定义的模型的输出(例如nn.Sequential
),另一个只是一个常数矩阵。如何将常量注入nn
的管道中,如果我这样做,我应该担心优化器会开始更改吗?
我知道我可以通过以下方式解决注射问题:
nn.Module
。这似乎很重要。nn.Module
调用的:forward(input)
子类,并允许消费者对常量的存在一无所知。nn.ParallelTable
,但这也会将常量暴露给模型消费者。nn.Linear
并覆盖权重。我只是不确定如何阻止优化程序执行更新。答案 0 :(得分:0)
您可以创建一个nn.Linear并覆盖:accGradParameters为无操作函数
m = nn.Linear(100,200)
-- copy your weights / bias into m.weight / m.bias
m.accGradParameters = function() end
-- m is a constant multiplier thing
答案 1 :(得分:0)
m=nn.MulConstant(7,true)(myMatrix)