tensorflow1 コードのアップグレード

TensorFlow1.x のコードをバージョン 2.x に移植します。

tensorflow2.x に含まれるツール tf_upgrade_v2 を使用できます。コードのアップグレードプロセスを簡単にします。
このツールは、単一ファイルの変換と複数ファイルのバッチ変換の両方をサポートしています。
1. 単一のコード ファイルを変換します
。 コマンド ラインに tf_upgrade_v2 コマンドを入力します。「-infile」パラメータは入力ファイルを指定し、「-outfile」パラメータは出力ファイルを指定します。コマンドは次のとおりです。

tf_upgrade_v2 --infile file_v1.py --outfile file_v2.py

2. 複数のコード ファイルを一括で変換する
また、tf_upgrade_v2 コマンドを入力します。「-intree」は入力ファイルのパス、「-outtree」は出力ファイルのパスを指定します。コマンドは次のとおりです:
tf_upgrade_v2 -intree file_v1 -outtree file_v2

また、この方法でほとんどの移植作業は解決できますが、一部の特殊な API については手動で移植する必要があります。
tf2 には tf.contrib モジュールがなくなり、アップグレード中に次の 2 つの部分に分割されます。

1. 1 つの部分は tf.2 のメイン フレームの下に移動され、tf.upgrade_v2 でアップグレードおよび変換できます。
2. 別の部品が削除されているため、変換できません。コードをアップグレードするときは、コードを手動で記述する必要があります。
さらに、tf1 で廃止されたアノテーションを持つ API は、tf2 には表示されません。変換に失敗したものは、tf2 が推奨する API に変換する必要があります。

おすすめ

転載: blog.csdn.net/qq_45919032/article/details/120462423