Function ForEachParam

nn/model.go:75–81

ForEachParam iterate all the parameters of a model also exploring the sub-parameters recursively.

(m Model, fn func(param *Param))

// ForEachParam iterate all the parameters of a model also exploring the sub-parameters recursively.
func ForEachParam(m Model, fn func(param *Param)) {
	paramsTraversal{
		paramsFunc:       fn,
		modelsFunc:       nil,
		exploreSubModels: true,
	}.walk(m)
}

// ForEachParamStrict iterate all the parameters of a model without exploring the sub-models.
func ForEachParamStrict(m Model, fn func(param *Param)) {