1. 程式人生 > >tensorflow程式碼舊版本0.x自動更新升級到1.0

tensorflow程式碼舊版本0.x自動更新升級到1.0

tensorflow 更新到1.0之後,0.n版本不相容,除了手動更改程式碼之外,tensorflow官方還提供了自動更新的指令碼。

使用方法:

  • 更新一個檔案:
    原本程式碼為foo.py, 使用tf_upgrade.py自動升級為1.0版本,新的檔名為foo-upgraded.py:
tf_upgrade.py --infile foo.py --outfile foo-upgraded.py
  • 目錄下的所有檔案都更新:
 tf_upgrade.py --intree InputDir --outtree OutputDir
  • 目錄下的所有檔案都更新,並複製除了python檔案之外的其他檔案到新資料夾:
    執行之後所有.py檔案都會更新並放在OutputDir目錄下,如果想要目錄中的其他檔案(.txt等)也複製到新的資料夾,可以設定copyotherfiles為True:
tf_upgrade.py --intree InputDir --outtree OutputDir --copyotherfiles True
  • 更新完畢後腳本會自動生成一個log檔案,其中包含了更新的內容。
third_party/tensorflow/tools/compatibility/test_file_v0.11.py Line 125

Renamed keyword argument from `dim` to `axis`
Renamed keyword argument from `squeeze_dims` to `axis`

    Old:                   [[1, 2, 3]]
, dim=1), squeeze_dims=[1]).eval(), ~~~~ ~~~~~~~~~~~~~ New: [[1, 2, 3]], axis=1), axis=[1]).eval(), ~~~~~ ~~~~~