sciai.common

Common module, involving datasets, initializers, optimizers, and train cells.

API Name

Description

Supported Platforms

sciai.common.DatasetGenerator

Common data generator.

GPU CPU Ascend

sciai.common.lbfgs_train

L-BFGS training function, which can only run on PYNATIVE mode currently.

GPU CPU Ascend

sciai.common.LbfgsOptimizer

L-BFGS second-order optimizer, which is currently only supported in PYNATIVE_MODE.

GPU CPU Ascend

sciai.common.LeCunNormal

Yann LeCun Normal Initialization \({N}(0, \text{sigma}^2)\) in order to initialize a tensor, where

GPU CPU Ascend

sciai.common.LeCunUniform

Yann LeCun Normal Initialization \({U}(-\text{boundary}, \text{boundary})\) in order to initialize a tensor, where

GPU CPU Ascend

sciai.common.Sampler

Common data sampler.

GPU CPU Ascend

sciai.common.StandardUniform

Generates an array with values sampled from Standard Uniform distribution \({U}(-\text{boundary}, \text{boundary})\) in order to initialize a tensor, where

GPU CPU Ascend

sciai.common.TrainCellWithCallBack

TrainOneStepCell with callbacks, which can handle multi-losses.

GPU CPU Ascend

sciai.common.TrainStepCell

Cell with gradient descent, similar to nn.TrainOneStepCell, but can accept multi-losses return.

Ascend GPU CPU

sciai.common.XavierTruncNormal

Xavier Truncated Normal Initialization with clip of 2 times of stddev from mean of Xavier Normal Initialization.

GPU CPU Ascend