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:
uIOMiZjkdW.png!large
TF2:
5MvPRel9Gz.png!large

重点

预定义 estimator、 自定义 estimatorFeature_column 组织数据。

本作品采用《CC 协议》,转载必须注明作者和本文链接

不要试图用百米冲刺的方法完成马拉松比赛。

讨论数量: 0
(= ̄ω ̄=)··· 暂无内容!

请勿发布不友善或者负能量的内容。与人为善,比聪明更重要!