sciai.common

常用模块包括数据集生成器、参数初始化器、优化器和训练器等。

接口名

概述

支持平台

sciai.common.DatasetGenerator

常用的数据生成器。

GPU CPU Ascend

sciai.common.lbfgs_train

L-BFGS训练函数,目前只能在PYNATIVE模式下运行。

GPU CPU Ascend

sciai.common.LbfgsOptimizer

L-BFGS 二阶优化器,目前仅在 PYNATIVE_MODE 中支持。

GPU CPU Ascend

sciai.common.LeCunNormal

生成一个服从Yan LeCun正态分布。

GPU CPU Ascend

sciai.common.LeCunUniform

生成一个服从Yan LeCun均匀分布。

GPU CPU Ascend

sciai.common.Sampler

常用的数据采样器。

GPU CPU Ascend

sciai.common.StandardUniform

生成一个服从标准均匀分布。

GPU CPU Ascend

sciai.common.TrainCellWithCallBack

带有回调的 TrainOneStepCell,可以处理多重损失。

GPU CPU Ascend

sciai.common.TrainStepCell

具有梯度下降的 Cell ,类似于 nn.TrainOneStepCell ,但可以接受多输出。

Ascend GPU CPU

sciai.common.XavierTruncNormal

Xavier截断正态初始化,丢弃Xavier正态初始化平均值附近2倍标准差外的点。

GPU CPU Ascend