sess.run API属于TensorFlow的低阶API,相对于Estimator来讲,灵活性较高,但模型的实现较为复杂。 TensorFlow 2.6版本中已经弃用该API,如果需要在TensorFlow 2.6版本中使用sess.run,需要通过compat.v1模块引用,方式如下: 1 tf.compat.v1.Session.run 使用sess.run API进行训练脚本开发的流程为: 数据预处理。 模型搭建/计算Loss/梯度更新。 创建session并初始化资源。 执行训练。 下面介绍如何迁移此类sess.run训练脚本,以便在AI处理器上进行训练。