tensorflow程式碼舊版本0.x自動更新升級到1.0
阿新 • • 發佈:2019-01-30
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(),
~~~~~ ~~~~~