前言
概述
当前业界基于TensorFlow的Python API开发的训练脚本,默认运行在CPU/GPU/TPU上。为了使其能够利用昇腾AI处理器的澎湃算力执行训练,提升训练性能,我们需要对训练网络脚本进行相关的迁移适配工作。当前昇腾AI处理器上支持TensorFlow的三种API开发的训练脚本迁移:分别是Estimator,Sess.run,Keras。
读者对象
本文档适用于AI算法工程师,用于将基于TensorFlow的Python API开发的训练脚本迁移到昇腾AI处理器上执行训练,并达到训练精度性能最优。
掌握以下经验和技能可以更好地理解本文档:
- 熟练的Python语言编程能力
- 熟悉TensorFlow 1.15的API
- 对机器学习、深度学习有一定的了解,熟悉训练网络的基本知识与流程
使用前须知
本文中的代码片段仅为示例,请用户使用时注意修改适配。