昇腾社区首页
中文
注册

训练场景介绍

mxRec提供使用TF.Session训练场景和NPUEstimator训练场景两种使用场景。

  • mxRec暂时不支持Keras。
  • mxRec目前仅支持模型的输入数据为tf.data.Dataset格式。

TF.Session训练场景

通过新建的Session实例启动模型运行,返回Tensor示例,进行定制化模型训练。

NPUEstimator训练场景

基于对机器学习不同阶段的控制的封装,用户无需不断的为新机器学习任务重复编写训练、评估、预测的代码,可以专注于对网络结构的控制。