Class TrainCfg

• Defined in File cfg.h

Class Documentation

class TrainCfg

Public Functions

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

obtain part of the name that identify a loss kernel.

返回

loss_name.

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

Set part of the name that identify a loss kernel.

参数

loss_name – [in] define part of the name that identify a loss kernel.