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.

Returns

loss_name.

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

Set part of the name that identify a loss kernel.

Parameters

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