Lines Matching refs:train_cfg
176 * @brief Destroy the train_cfg object. Only valid for Lite Train.
177 * @param train_cfg TrainCfg object handle.
180 OH_AI_API void OH_AI_TrainCfgDestroy(OH_AI_TrainCfgHandle *train_cfg);
184 * @param train_cfg TrainCfg object handle.
189 OH_AI_API char **OH_AI_TrainCfgGetLossName(OH_AI_TrainCfgHandle train_cfg, size_t *num);
193 * @param train_cfg TrainCfg object handle.
198 OH_AI_API void OH_AI_TrainCfgSetLossName(OH_AI_TrainCfgHandle train_cfg, const char **loss_name, size_t num);
201 * @brief Obtains optimization level of the train_cfg. Only valid for Lite Train.
202 * @param train_cfg TrainCfg object handle.
206 OH_AI_API OH_AI_OptimizationLevel OH_AI_TrainCfgGetOptimizationLevel(OH_AI_TrainCfgHandle train_cfg);
209 * @brief Set optimization level of the train_cfg. Only valid for Lite Train.
210 * @param train_cfg TrainCfg object handle.
211 * @param level The optimization level of train_cfg.
214 OH_AI_API void OH_AI_TrainCfgSetOptimizationLevel(OH_AI_TrainCfgHandle train_cfg, OH_AI_OptimizationLevel level);
223 * @param train_cfg Define the config used by training.
229 const OH_AI_TrainCfgHandle train_cfg);
237 * @param train_cfg Define the config used by training.
244 const OH_AI_TrainCfgHandle train_cfg);