接口名 |
概述 |
支持平台 |
mindflow.core.AdaHessian
|
二阶优化器 AdaHessian，利用 Hessian 矩阵对角元信息进行二阶优化求解。 |
Ascend
|
mindflow.core.get_multi_step_lr
|
epoch的数量到达其中一个milestone时，学习率按 gamma 进行衰减，生成学习率数组。 |
Ascend GPU CPU
|
mindflow.core.get_poly_lr
|
生成指数衰减的学习率。 |
Ascend GPU
|
mindflow.core.get_warmup_cosine_annealing_lr
|
基于余弦函数生成衰减学习率数组。 |
Ascend GPU CPU
|
mindflow.core.MTLWeightedLoss
|
计算MTL策略自动加权多任务损失。 |
Ascend GPU
|
mindflow.core.RelativeRMSELoss
|
相对均方根误差（RRMSE）是由均方根值归一化的均方根误差，其中每个残差都是根据实际值缩放的。 |
Ascend GPU CPU
|
mindflow.core.WaveletTransformLoss
|
多级小波变换损失。 |
Ascend GPU
|
mindflow.core.get_loss_metric
|
获取损失函数。 |
Ascend GPU
|
mindflow.core.batched_hessian
|
计算网络模型的海森矩阵。 |
Ascend GPU CPU
|
mindflow.core.batched_jacobian
|
计算网络模型的雅可比矩阵。 |
Ascend GPU CPU
|