mx/optimizer.go
exported function AdamUpdate should have comment or be unexported Open
Open
func AdamUpdate(params, grads, mean, variance *NDArray, lr, beta1, beta2, epsilon, wd float64) {
- Exclude checks
exported function SgdMomUpdate should have comment or be unexported Open
Open
func SgdMomUpdate(params, grads, mom *NDArray, lr, momentum, wd float64) {
- Exclude checks
exported function SgdUpdate should have comment or be unexported Open
Open
func SgdUpdate(params, grads *NDArray, lr, wd float64) {
- Exclude checks