TF1 程式碼升級 TF2 思路

Galois發表於2020-03-10

如何升級

  • 替換 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 協議》,轉載必須註明作者和本文連結

不要試圖用百米衝刺的方法完成馬拉松比賽。

相關文章