Function MSTrainCfgSetLossName

Function Documentation

void MSTrainCfgSetLossName(MSTrainCfgHandle train_cfg, const char **loss_name, size_t num)

Set part of the name that identify a loss kernel. Only valid for Lite Train.

Parameters
  • train_cfg[in] TrainCfg object handle.

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

  • num[in] The num of loss_name.