From abe3d0bfde141cfc04d7a33f4be1cf14aac2a760 Mon Sep 17 00:00:00 2001 From: yuhan Date: Thu, 24 Apr 2025 13:51:23 +0800 Subject: [PATCH] modify mindflow.core.AdaHessian params --- docs/api_python/mindflow/core/mindflow.core.AdaHessian.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/api_python/mindflow/core/mindflow.core.AdaHessian.rst b/docs/api_python/mindflow/core/mindflow.core.AdaHessian.rst index 6b75f82c0..ee3a3c9e4 100644 --- a/docs/api_python/mindflow/core/mindflow.core.AdaHessian.rst +++ b/docs/api_python/mindflow/core/mindflow.core.AdaHessian.rst @@ -1,7 +1,7 @@ mindflow.core.AdaHessian ========================= -.. py:class:: mindflow.core.AdaHessian(*args, **kwargs) +.. py:class:: mindflow.core.AdaHessian(params, learning_rate=1e-3, beta1=0.9, beta2=0.999, eps=1e-8, use_locking=False, use_nesterov=False, weight_decay=0.0, loss_scale=1.0, use_amsgrad=False, **kwargs) 二阶优化器 AdaHessian，利用 Hessian 矩阵对角元信息进行二阶优化求解。 有关更多详细信息，请参考论文 `ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learning `_ 。 -- Gitee