從 TensorFlow 1.x 遷移至 TensorFlow 2

瞭解如何將 TensorFlow 程式碼從 TensorFlow 1.x 遷移至 TensorFlow 2。轉換程式碼可能需要一些工夫,但每次變更都能讓您使用新功能和模型、提高清晰度和簡潔性,並更輕鬆地進行偵錯。在開始遷移之前,請先閱讀行為指南。簡而言之,遷移程序如下:

  1. 執行自動化指令碼,將您的 TF1.x API 用法轉換為 tf.compat.v1
  2. 移除舊的 tf.contrib.layers,並將其替換為 TF Slim 符號。另請查看 TF Addons 以取得其他 tf.contrib 符號。
  3. 重新編寫您的 TF1.x 模型正向傳遞,使其在 TF2 中執行,並啟用 Eager Execution。
  4. 驗證遷移程式碼的準確性和數值正確性
  5. 將您的 訓練、評估模型儲存程式碼升級至 TF2 對等項目。
  6. (選用) 遷移您的 TF2 相容 tf.compat.v1 API (包括 TF Slim 用法) 至慣用的 TF2 API。
瞭解 TF2 API 和行為與 TF1.x 的根本差異。
立即開始在 TF2 中使用 TF1.x 模型,方法是使用模型化墊片。
以程式設計方式將 TF1.x 程式碼的某些部分升級至 TF2。
驗證遷移的 TF2 程式碼是否正確。
從您的估算器訓練管線遷移至 TF2。
瞭解如何從 tf.feature_columns 遷移至 Keras 預處理層。
瞭解如何將多工作站分散式 Estimator 遷移至 TF2。
瞭解如何將 TPUEstimator API 遷移至 TF2。
瞭解如何將使用 TF1.x 建立的 TF Lite 程式碼遷移至 TF2。