Class TrainCfg

#include <cfg.h>

TrainCfgMindSpore Lite训练的相关配置参数。

构造函数

TrainCfg() { this->loss_name_ = "_loss_fn"; }

TrainCfg(const TrainCfg &rhs) {
    this->loss_name_ = rhs.loss_name_;
    this->mix_precision_cfg_ = rhs.mix_precision_cfg_;
    this->accumulate_gradients_ = rhs.accumulate_gradients_;
  }

• 参数

  • rhs: 训练配置。

析构函数

~TrainCfg() = default

公有成员变量

OptimizationLevel optimization_level_ = kO0

优化的数据类型。

enum OptimizationLevel : uint32_t {
  kO0 = 0,
  kO2 = 2,
  kO3 = 3,
  kAuto = 4,
  kOptimizationType = 0xFFFFFFFF
};

std::string loss_name_

损失节点的名称。

MixPrecisionCfg mix_precision_cfg_

混合精度配置。

bool accumulate_gradients_

是否累加梯度。

公有成员函数

GetLossName

inline std::vector<std::string> GetLossName() const

获得损失名称。

• 返回值

  损失的名称。

SetLossName

inline void SetLossName(const std::vector<std::string> &loss_name)

设置损失名称。

• 参数

  • loss_name: 损失的名称。