TrainCfg

import com.mindspore.config.TrainCfg;

用于端上模型训练的配置参数。

公有成员函数

function	云侧推理是否支持	端侧推理是否支持
boolean init()	✕	√
boolean init(String loss_name)	✕	√
void free()	✕	√
boolean addMixPrecisionCfg(boolean dynamicLossScale, float lossScale, int thresholdIterNum)	✕	√
long getTrainCfgPtr()	✕	√

init

public boolean init()

初始化训练配置。

• 返回值

  初始化状态。

public boolean init(String loss_name)

初始化训练配置指定损失函数名称。

• 参数

• loss_name:用于分割推理和训练部分的损失函数名称。

• 返回值

  初始化状态。

free

public void free()

释放训练配置。

addMixPrecisionCfg

public boolean addMixPrecisionCfg(boolean dynamicLossScale, float lossScale, int thresholdIterNum)

将混合精度配置添加到训练配置中。

• 参数

• dynamicLossScale: 是动态还是静态损失比例因子。

• lossScale:损失比例因子 。

• thresholdIterNum: 启用dynamicLossScale时修改损失比例因子的阈值。

• 返回值

  添加状态。

getTrainCfgPtr

public long getTrainCfgPtr()

获取训练配置指针。

• 返回值

  训练配置指针。