addParameter

open override fun addParameter(param: Parameter, applyWeightDecay: Boolean)(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


open override fun addParameter(param: ModuleParameter<*, *>, applyWeightDecay: Boolean)(source)

Register a raw module parameter to be optimized.