Update godoc

itsubaki · Oct 17, 2023 · 22438e2 · 22438e2
1 parent 146de46
commit 22438e2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/optimizer/hook/weight_decay.go b/optimizer/hook/weight_decay.go
@@ -2,7 +2,7 @@ package hook
 
 import "github.com/itsubaki/neu/math/matrix"
 
-// Decay returns a function that applies weight decay to the gradients.
+// WeightDecay returns a function that applies weight decay to the gradients.
 func WeightDecay(lambda float64) func(params, grads [][]matrix.Matrix) [][]matrix.Matrix {
 	return func(params, grads [][]matrix.Matrix) [][]matrix.Matrix {
 		out := make([][]matrix.Matrix, len(params))