skainet-lang-core/sk.ainet.lang.nn.optim/Optimizer/addParameter

addParameter

abstract fun addParameter(param: Parameter, applyWeightDecay: Boolean = true)(source)

Register a parameter to be optimized.

Parameters

param

module parameter to update during step

applyWeightDecay

whether to apply weight decay to this parameter when weightDecay> 0

abstract fun addParameter(param: ModuleParameter<*, *>, applyWeightDecay: Boolean = true)(source)

Register a raw module parameter to be optimized.