训练场景介绍
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训练场景
基于对机器学习不同阶段的控制的封装,用户无需不断的为新机器学习任务重复编写训练、评估、预测的代码,可以专注于对网络结构的控制。
父主题: 模型训练介绍