下载
中文
注册

训练场景介绍

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

  • Rec SDK暂时不支持Keras。
  • 目前Rec SDK仅支持使用Tensorflow原生API模型训练脚本迁移,不支持使用第三方框架(tf_adapter、HugeCTR、DeepRec等)。
  • Rec SDK目前仅支持模型的输入数据为tf.data.Dataset格式。
  • 启用大小循环的情况下,训练迭代的总次数必须是小循环(即iterations_per_loop)的整数倍。

TF.Session训练场景

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

NPUEstimator训练场景

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