TF1 代码升级 TF2 思路
如何升级
- 替换 Seession.run
- feed_dict、tf.placeholder 变成函数调用
- 替换 API
- tf.get_variable 替换为 tf.Variable
- variable_scope「区分变量」 被替换为以下东西的一个:
- tf.keras.layers.Layer
- tf.keras.Model
- tf.Module
- 升级训练流程
- 使用 tf.keras.Model.fit
- 升级数据输入
- Iterator 变成直接输入
Tf1 & TF2 代码对比
TF1:
TF2:
重点
预定义 estimator
、 自定义 estimator
、Feature_column
组织数据。
本作品采用《CC 协议》,转载必须注明作者和本文链接