sciai.common
常用模块包括数据集生成器、参数初始化器、优化器和训练器等。
接口名 |
概述 |
支持平台 |
常用的数据生成器。 |
|
|
L-BFGS训练函数,目前只能在PYNATIVE模式下运行。 |
|
|
L-BFGS 二阶优化器,目前仅在 PYNATIVE_MODE 中支持。 |
|
|
生成一个服从Yan LeCun正态分布。 |
|
|
生成一个服从Yan LeCun均匀分布。 |
|
|
常用的数据采样器。 |
|
|
生成一个服从标准均匀分布。 |
|
|
带有回调的 TrainOneStepCell,可以处理多重损失。 |
|
|
具有梯度下降的 Cell ,类似于 nn.TrainOneStepCell ,但可以接受多输出。 |
|
|
Xavier截断正态初始化,丢弃Xavier正态初始化平均值附近2倍标准差外的点。 |
|